Searched defs:swizzle (Results 1 – 6 of 6) sorted by relevance
| /llvm-project/mlir/lib/CAPI/Dialect/ |
| H A D | NVGPU.cpp | 25 int swizzle, int l2promo, in mlirNVGPUTensorMapDescriptorTypeGet()
|
| /llvm-project/mlir/lib/Bindings/Python/ |
| H A D | DialectNVGPU.cpp | 27 __anonc1027f100102(py::object cls, MlirType tensorMemrefType, int swizzle, int l2promo, int oobFill, int interleave, MlirContext ctx) populateDialectNVGPUSubmodule() argument
|
| /llvm-project/mlir/test/Integration/GPU/CUDA/sm90/python/tools/ |
| H A D | matmulBuilder.py | 34 def __init__(self, swizzle, l2promo, oob, interleave, tma_box_shape, memref_ty): argument
|
| /llvm-project/mlir/test/Examples/NVGPU/tools/ |
| H A D | nvdsl.py | 109 swizzle=nvgpu.TensorMapSwizzleKind.SWIZZLE_NONE, argument
|
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | CudaRuntimeWrappers.cpp | 402 mgpuTensorMapEncodeTiled(CUtensorMap * tensorMap,CUtensorMapDataType tensorDataType,cuuint32_t tensorRank,void * globalAddress,const cuuint64_t * globalDim,const cuuint64_t * globalStrides,const cuuint32_t * boxDim,const cuuint32_t * elementStrides,CUtensorMapInterleave interleave,CUtensorMapSwizzle swizzle,CUtensorMapL2promotion l2Promotion,CUtensorMapFloatOOBfill oobFill) mgpuTensorMapEncodeTiled() argument
454 mgpuTensorMapEncodeTiledMemref(int64_t tensorRank,void * rankedDescriptor,const CUtensorMapDataType tensorDataType,CUtensorMapInterleave interleave,CUtensorMapSwizzle swizzle,CUtensorMapL2promotion l2Promotion,CUtensorMapFloatOOBfill oobFill,int64_t * inputBoxDims) mgpuTensorMapEncodeTiledMemref() argument
|
| /llvm-project/mlir/lib/Conversion/NVGPUToNVVM/ |
| H A D | NVGPUToNVVM.cpp | 1041 unsigned swizzle = matchAndRewrite() local
|