github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-15 00:55:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

rocm path in autogen (#3697)

Browse Source
This commit is contained in:
nimlgen
2024-03-12 14:06:43 +03:00
committed by GitHub
parent 971d7f5d7c
commit dd1a1c12df
4 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/runtime/autogen/comgr.py
View File

@@ -6,7 +6,7 @@
# POINTER_SIZE is: 8
# LONGDOUBLE_SIZE is: 16
#
import ctypes
import ctypes, os
def string_cast(char_pointer, encoding='utf-8', errors='strict'):
@@ -29,7 +29,7 @@ def char_pointer_cast(string, encoding='utf-8'):
_libraries = {}
_libraries['libamd_comgr.so'] = ctypes.CDLL('/opt/rocm/lib/libamd_comgr.so')
_libraries['libamd_comgr.so'] = ctypes.CDLL(os.getenv('ROCM_PATH', '/opt/rocm/')+'/lib/libamd_comgr.so')
c_int128 = ctypes.c_ubyte*16
c_uint128 = c_int128
void = None

4
tinygrad/runtime/autogen/hip.py
View File

@@ -6,7 +6,7 @@
# POINTER_SIZE is: 8
# LONGDOUBLE_SIZE is: 16
#
import ctypes
import ctypes, os
class AsDictMixin:
@@ -155,7 +155,7 @@ def char_pointer_cast(string, encoding='utf-8'):
_libraries['libamdhip64.so'] = ctypes.CDLL('/opt/rocm/lib/libamdhip64.so')
_libraries['libamdhip64.so'] = ctypes.CDLL(os.getenv('ROCM_PATH', '/opt/rocm/')+'/lib/libamdhip64.so')

4
tinygrad/runtime/autogen/hsa.py
View File

@@ -6,7 +6,7 @@
# POINTER_SIZE is: 8
# LONGDOUBLE_SIZE is: 16
#
import ctypes
import ctypes, os
def string_cast(char_pointer, encoding='utf-8', errors='strict'):
@@ -29,7 +29,7 @@ def char_pointer_cast(string, encoding='utf-8'):
_libraries = {}
_libraries['libhsa-runtime64.so'] = ctypes.CDLL('/opt/rocm/lib/libhsa-runtime64.so')
_libraries['libhsa-runtime64.so'] = ctypes.CDLL(os.getenv('ROCM_PATH', '/opt/rocm/')+'/lib/libhsa-runtime64.so')
class AsDictMixin:
@classmethod
def as_dict(cls, self):