Skip to content

Commit 486903b

Browse files
authored
Ufuncs: allow passing string for kokkos profiler kernel name for some ufuncs from npbench (#287)
1 parent 5d645be commit 486903b

3 files changed

Lines changed: 111 additions & 69 deletions

File tree

pykokkos/interface/parallel_dispatch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def handle_args(is_for: bool, *args) -> HandledArgs:
5050
workunit = unpacked[1]
5151

5252
elif len(unpacked) == 3:
53-
if isinstance(unpacked[0], str):
53+
if isinstance(unpacked[0], str) or unpacked[0] is None:
5454
name = unpacked[0]
5555
policy = unpacked[1]
5656
workunit = unpacked[2]
@@ -66,7 +66,7 @@ def handle_args(is_for: bool, *args) -> HandledArgs:
6666
raise TypeError(f"ERROR: wrong arguments {unpacked}")
6767

6868
elif len(unpacked) == 4:
69-
if isinstance(unpacked[0], str):
69+
if isinstance(unpacked[0], str) or unpacked[0] is None:
7070
name = unpacked[0]
7171
policy = unpacked[1]
7272
workunit = unpacked[2]

0 commit comments

Comments
 (0)