Pass the architecture and warp size to the HIP backend.

PyFR · Feb 26, 2023 · f3e7d90 · f3e7d90
1 parent af42472
commit f3e7d90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gimmik/hip.py b/gimmik/hip.py
@@ -7,7 +7,7 @@ class HIPMatMul(MatMul):
     platform = 'hip'
     basemeta = {'block': (128, 1, 1), 'width': 1, 'shared': 0}
 
-    def _kernel_generators(self, dtype, dsize):
+    def _kernel_generators(self, dtype, dsize, *, gcn_arch=None, warp_size=64):
         # B loading, C streaming kernel
         yield ('cstream', {}, {})