[Mlir-commits] [mlir] 2379432 - [MLIR] Correct Initial TMA Descriptor Values (#67397)

[llvmlistbot at llvm.org]

Author: Guray Ozen
Date: 2023-09-26T09:20:46+02:00
New Revision: 2379432c8844c16c33f32bfc60542a89e7e830dc

URL: https://github.com/llvm/llvm-project/commit/2379432c8844c16c33f32bfc60542a89e7e830dc
DIFF: https://github.com/llvm/llvm-project/commit/2379432c8844c16c33f32bfc60542a89e7e830dc.diff

LOG: [MLIR] Correct Initial TMA Descriptor Values (#67397)

Added: 
    

Modified: 
    mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp b/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
index 3dbbc674128732e..2c04821303a49f9 100644
--- a/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
+++ b/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
@@ -376,8 +376,8 @@ extern "C" MLIR_CUDA_WRAPPERS_EXPORT void *mgpuTensorMapEncodeTiledMemref(
   CUtensorMap tensorMap;
 
   auto *globalAddress = descriptor->data;
-  uint32_t boxDim[5] = {0}, elementStrides[5] = {0};
-  uint64_t globalDim[5] = {0}, globalStrides[5] = {0};
+  uint32_t boxDim[5] = {1, 1, 1, 1, 1}, elementStrides[5] = {1, 1, 1, 1, 1};
+  uint64_t globalDim[5] = {1, 1, 1, 1, 1}, globalStrides[5] = {0};
   uint32_t tensorRank32 = uint32_t(tensorRank);
 
   static const int elementSizeInBytes[] = {1, 2, 4, 4, 8, 8, 2,