You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
12 lines
430 B
12 lines
430 B
1 month ago
|
import ctypes
|
||
|
|
import tinygrad.runtime.autogen.hip as hip
|
||
|
|
from tinygrad.runtime.ops_hip import check
|
||
|
|
from tinygrad.helpers import init_c_var
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
check(hip.hipSetDevice(0))
|
||
|
|
evt = init_c_var(hip.hipEvent_t(), lambda x: check(hip.hipEventCreate(ctypes.byref(x))))
|
||
|
|
check(hip.hipSetDevice(1))
|
||
|
|
check(hip.hipStreamWaitEvent(None, evt, 0))
|
||
|
|
check(hip.hipSetDevice(0))
|
||
|
|
check(hip.hipEventRecord(evt, None))
|