Add additional implicit casts for dim3

CHIP-SPV · Jul 27, 2024 · 32a6426 · 32a6426
1 parent 142496a
commit 32a6426
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/include/hip/hip_runtime_api.h b/include/hip/hip_runtime_api.h
@@ -1069,7 +1069,13 @@ typedef struct dim3 {
     uint32_t y;  ///< y
     uint32_t z;  ///< z
 #ifdef __cplusplus
-    constexpr __host__ __device__ dim3(uint32_t _x = 1, uint32_t _y = 1, uint32_t _z = 1) : x(_x), y(_y), z(_z){};
+    constexpr __host__ __device__ dim3(uint32_t _x = 1, uint32_t _y = 1, uint32_t _z = 1) : x(_x), y(_y), z(_z) {}
+#endif
+#if defined(__clang__) && defined(__HIP__)
+    __host__ __device__ dim3(const __hip_builtin_threadIdx_t& idx) : x(idx.x), y(idx.y), z(idx.z) {}
+    __host__ __device__ dim3(const __hip_builtin_blockIdx_t& idx) : x(idx.x), y(idx.y), z(idx.z) {}
+    __host__ __device__ dim3(const __hip_builtin_blockDim_t& dim) : x(dim.x), y(dim.y), z(dim.z) {}
+    __host__ __device__ dim3(const __hip_builtin_gridDim_t& dim) : x(dim.x), y(dim.y), z(dim.z) {}
 #endif
 } dim3;
 /**