Skip to content
Snippets Groups Projects
Commit 08efb35e authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Don't error out for unexpected dtype sizes in get_simd_group_size().

parent 1cb9f09b
No related branches found
No related tags found
No related merge requests found
...@@ -297,6 +297,9 @@ def get_simd_group_size(dev, type_size): ...@@ -297,6 +297,9 @@ def get_simd_group_size(dev, type_size):
elif type_size == 8: elif type_size == 8:
return dev.preferred_vector_width_double return dev.preferred_vector_width_double
else: else:
raise ValueError("unexpected dtype size in get_simd_group_size") from warnings import warn
warn("unexpected dtype size in get_simd_group on CPU device, "
"guessing group width 1")
return 1
return None return None
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment