Searched defs:globalDim (Results 1 – 1 of 1) sorted by relevance
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | CudaRuntimeWrappers.cpp | 397 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
433 mgpuGetMemRefDataAndShape(void * rawDescriptor,char ** addr,uint64_t * globalDim,uint64_t * globalStrides,const CUtensorMapDataType tensorDataType) mgpuGetMemRefDataAndShape() argument
462 uint64_t globalDim[5] = {1, 1, 1, 1, 1}, globalStrides[5] = {0}; mgpuTensorMapEncodeTiledMemref() local
[all...] |