From f3e7d900b43ba5f403dbc98206fea6cfdae5b0fc Mon Sep 17 00:00:00 2001 From: Freddie Witherden Date: Sun, 26 Feb 2023 17:22:19 -0600 Subject: [PATCH] Pass the architecture and warp size to the HIP backend. --- gimmik/hip.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gimmik/hip.py b/gimmik/hip.py index 4e6a210..a216bd3 100644 --- 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', {}, {})