Searched defs:interleave (Results 1 – 6 of 6) sorted by relevance
| /llvm-project/mlir/lib/CAPI/Dialect/ |
| H A D | NVGPU.cpp | 26 int oobFill, int interleave) { in mlirNVGPUTensorMapDescriptorTypeGet()
|
| /llvm-project/mlir/lib/Bindings/Python/ |
| H A D | DialectNVGPU.cpp | 27 int oobFill, int interleave, MlirContext ctx) { in populateDialectNVGPUSubmodule() argument
|
| /llvm-project/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | LowerVectorInterleave.cpp | 68 Value interleave = in matchAndRewrite() local
|
| /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 | 112 interleave=nvgpu.TensorMapInterleaveKind.INTERLEAVE_NONE, argument
|
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | CudaRuntimeWrappers.cpp | 401 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
453 mgpuTensorMapEncodeTiledMemref(int64_t tensorRank,void * rankedDescriptor,const CUtensorMapDataType tensorDataType,CUtensorMapInterleave interleave,CUtensorMapSwizzle swizzle,CUtensorMapL2promotion l2Promotion,CUtensorMapFloatOOBfill oobFill,int64_t * inputBoxDims) mgpuTensorMapEncodeTiledMemref() argument
|