Guray Ozen e56d6745f7 [mlir][nvgpu] Add tma.create.descriptor to create tensor map descriptor ...

The Op creates a tensor map descriptor object representing tiled memory region. The descriptor is used by Tensor Memory Access (TMA). The `tensor` is the source tensor to be tiled. The `boxDimensions` is the size of the tiled memory region in each dimension.

The pattern here lowers `tma.create.descriptor` to a runtime function call that eventually calls calls CUDA Driver's `cuTensorMapEncodeTiled`. For more information see below:
https://docs.nvidia.com/cuda/cuda-driver-api/group__CUDA__TENSOR__MEMORY.html

Depends on D155453

Reviewed By: nicolasvasilache

Differential Revision: https://reviews.llvm.org/D155680

2023-07-21 11:33:04 +02:00

..

mlir

[mlir][nvgpu] Add tma.create.descriptor to create tensor map descriptor

2023-07-21 11:33:04 +02:00

mlir-c

[MLIR][CAPI] Add C API dialect registration methods for Arith, Math, MemRef and Vector dialects

2023-07-17 14:45:49 +00:00