Fix 12.9 compatibility

2025-05-07 13:23:40 +08:00
parent 085b4a1532
commit 8702f910e3
4 changed files with 24 additions and 20 deletions
--- a/tests/test_jit.py
+++ b/tests/test_jit.py
@@ -39,7 +39,9 @@ __global__ void vector_add(T* a, T* b, T* c, uint32_t n) {{
    }}
 }}

-auto ptr = reinterpret_cast<void*>(&vector_add<float>);
+static void __instantiate_kernel() {{
+    auto ptr = reinterpret_cast<void*>(&vector_add<{kwargs['T']}>);
+}}
 """

    # noinspection PyShadowingNames,PyMethodOverriding