From 7722d2ee0c7677ced14269cbd7bd8c56dc3ca560 Mon Sep 17 00:00:00 2001
From: Nora Abi Akar <nora.abiakar@gmail.com>
Date: Thu, 19 Aug 2021 17:24:48 +0200
Subject: [PATCH] Add dim3 to gridDim constructor in kernel launch. (#1630)
Fixes hipcc rocm-4.0.1 build for AMD GPU
Fixes #1628
---
modcc/printer/gpuprinter.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modcc/printer/gpuprinter.cpp b/modcc/printer/gpuprinter.cpp
index 00dd9e7f..5343f023 100644
--- a/modcc/printer/gpuprinter.cpp
+++ b/modcc/printer/gpuprinter.cpp
@@ -303,7 +303,7 @@ std::string emit_gpu_cu_source(const Module& module_, const printer_options& opt
" unsigned grid_dim = ::arb::gpu::impl::block_count(n, block_dim);\n"
" {1}<<<grid_dim, block_dim>>>(*p);\n"
" if (!p->multiplicity) return;\n"
- " multiply<<<{{grid_dim, {2}}}, block_dim>>>(*p);\n"),
+ " multiply<<<dim3{{grid_dim, {2}}}, block_dim>>>(*p);\n"),
"width",
api_name,
n);
--
GitLab