smyalygames/FiniteVolumeGPU
1
0
Fork 0
mirror of https://github.com/smyalygames/FiniteVolumeGPU.git synced 2025-10-31 20:17:41 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(kernel): add environment condition for importing certain packages

Browse Source
This commit is contained in:
Anthony Berg 2025-06-24 19:57:59 +02:00
parent 3b424d1f5a
commit d4f2ffc493
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
GPUSimulators/gpu/__init__.py
View File

@ -1,2 +1,8 @@
from .cuda_context import CudaContext
from .hip_context import HIPContext
from os import environ
__env_name = 'GPU_LANG'
if __env_name in environ and environ.get(__env_name).lower() == "cuda":
from .cuda_context import CudaContext as KernelContext
else:
from .hip_context import HIPContext as KernelContext