From 81f7c6800aa66ca79aa016eb09b83cd825223e0c Mon Sep 17 00:00:00 2001 From: calledit <1573053+calledit@users.noreply.github.com> Date: Thu, 23 Feb 2023 22:59:02 +0100 Subject: [PATCH] Added info on simdgroup availability (#586) * Add info on simdgroup availability * "osx" not "os x" * Update metal_matmul.py * Update metal_matmul.py --- extra/gemm/metal_matmul.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extra/gemm/metal_matmul.py b/extra/gemm/metal_matmul.py index 1b7b80a8b0..7d2839ad89 100644 --- a/extra/gemm/metal_matmul.py +++ b/extra/gemm/metal_matmul.py @@ -25,7 +25,7 @@ FLOPS = N*N*N*2 prog = CLProgram("test", f""" #include -#include +#include // Available from Metal version 2.3 released with OS X 11.0+ using namespace metal; kernel void test(device float *a, device const float *data1, device const float *data2, uint3 gid [[thread_position_in_grid]], uint3 xid [[threadgroup_position_in_grid]], uint3 lid [[thread_position_in_threadgroup]], uint sidx [[simdgroup_index_in_threadgroup]]) {{ // 1-2 simd groups