Ufuncs: allow passing string for kokkos profiler kernel name for some…

… ufuncs from npbench (#287)
kokkos · Aug 28, 2024 · 486903b · 486903b
1 parent 5d645be
commit 486903b
Show file tree

Hide file tree

Showing 3 changed files with 111 additions and 69 deletions.
diff --git a/pykokkos/interface/parallel_dispatch.py b/pykokkos/interface/parallel_dispatch.py
@@ -50,7 +50,7 @@ def handle_args(is_for: bool, *args) -> HandledArgs:
         workunit = unpacked[1]
 
     elif len(unpacked) == 3:
-        if isinstance(unpacked[0], str):
+        if isinstance(unpacked[0], str) or unpacked[0] is None:
             name = unpacked[0]
             policy = unpacked[1]
             workunit = unpacked[2]
@@ -66,7 +66,7 @@ def handle_args(is_for: bool, *args) -> HandledArgs:
             raise TypeError(f"ERROR: wrong arguments {unpacked}")
 
     elif len(unpacked) == 4:
-        if isinstance(unpacked[0], str):
+        if isinstance(unpacked[0], str) or unpacked[0] is None:
             name = unpacked[0]
             policy = unpacked[1]
             workunit = unpacked[2]