github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-14 16:44:59 -05:00
Code Issues Packages Projects Releases Wiki Activity

hsa in lin opts (#3602)

Browse Source
This commit is contained in:
nimlgen
2024-03-04 17:17:32 +03:00
committed by GitHub
parent 7c90005c65
commit 3db826e195
3 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/runtime/ops_hsa.py
View File

@@ -4,10 +4,12 @@ from typing import Tuple, TypeVar, List, Dict
import tinygrad.runtime.autogen.hsa as hsa
from tinygrad.helpers import DEBUG, init_c_var, from_mv, round_up, to_mv, init_c_struct_t
from tinygrad.device import Compiled, LRUAllocator, BufferOptions
from tinygrad.codegen.kernel import LinearizerOptions
from tinygrad.runtime.ops_hip import HIPCompiler
from tinygrad.runtime.driver.hsa import check, scan_agents, find_memory_pool, AQLQueue
HSACompiler = HIPCompiler
class HSACompiler(HIPCompiler):
linearizer_opts = LinearizerOptions("HSA", has_tensor_cores=True)
class HSAProgram:
def __init__(self, device:HSADevice, name:str, lib:bytes):