Skip to content
Snippets Groups Projects

Various extensions to the vectorizer

Merged Daniel Bauer requested to merge hyteg/pystencils:bauerd/vec-extensions into v2.0-dev
Viewing commit 75210689
Show latest version
2 files
+ 106
50
Preferences
Compare changes
Files
2
@@ -115,6 +115,18 @@ class Target(Flag):
return avail_targets.pop()
else:
return Target.GenericCPU
@staticmethod
def available_targets() -> list[Target]:
targets = [Target.GenericCPU]
try:
import cupy # noqa: F401
targets.append(Target.CUDA)
except ImportError:
pass
targets += Target.available_vector_cpu_targets()
return targets
@staticmethod
def available_vector_cpu_targets() -> list[Target]: