mirror of
https://github.com/smyalygames/FiniteVolumeGPU.git
synced 2025-11-27 23:09:49 +01:00
8 lines
242 B
Python
8 lines
242 B
Python
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 |