#2716 mark MATMUL as available on GPU

stfc · Oct 8, 2024 · 721ff5d · 721ff5d
1 parent 2e76d3a
commit 721ff5d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/psyclone/psyir/nodes/intrinsic_call.py b/src/psyclone/psyir/nodes/intrinsic_call.py
@@ -797,9 +797,10 @@ def is_available_on_device(self):
             IntrinsicCall.Intrinsic.SIGN, IntrinsicCall.Intrinsic.SIN,
             IntrinsicCall.Intrinsic.SINH, IntrinsicCall.Intrinsic.SQRT,
             IntrinsicCall.Intrinsic.TAN, IntrinsicCall.Intrinsic.TANH,
-            # The one below are not documented on nvidia compiler
+            # The ones below are not documented in the nvidia compiler docs
             IntrinsicCall.Intrinsic.PRODUCT, IntrinsicCall.Intrinsic.SIZE,
             IntrinsicCall.Intrinsic.SUM, IntrinsicCall.Intrinsic.LBOUND,
+            IntrinsicCall.Intrinsic.MATMUL,
             IntrinsicCall.Intrinsic.UBOUND)
 
     @classmethod