feat(common): add imports for HIP 2d arrays

This commit is contained in:
Anthony Berg 2025-06-25 13:12:55 +02:00
parent 01b39cc9b4
commit c549556c36

View File

@ -1,11 +1,10 @@
from os import environ
# TODO this is temporary, remove
from .cuda import *
__env_name = 'GPU_LANG'
if __env_name in environ and environ.get(__env_name).lower() == "cuda":
from .cuda import *
else:
from .hip import *
from .hip import *
# TODO this is temporary, remove
from .cuda import array3d