Code

[NDArray] Map NDArray ops to GPUs#974

Code

Merged

fschlimb merged 5 commits intointel:mainintel/mlir-extensions:mainfrom tkarna:ndarray-gpu-upstreamtkarna/mlir-extensions:ndarray-gpu-upstreamCopy head branch name to clipboardDec 17, 2024

+564-9

Commits

Commits on Dec 3, 2024

• NDArrayToLinalg: mark ndarray.copy with region env if it involves a gpu array

  tkarna

  committed
  06cb9f1

• InsertGPUAllocs: add host-shared option, defaults to true

  tkarna

  committed
  77774a6

Commits on Dec 17, 2024

• Add insert-gpu-copy pass

  tkarna

  committed
  55f55a5

• add convert-region-parallel-loops-to-gpu pass converting parloops only in gpu regions

  Show description for 371a6b4
  

  fschlimb

  authored and

  tkarna

  committed
  371a6b4

• Add tile-loops pass

  tkarna

  committed
  412577e