mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-13 00:15:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

the compiler is a driver (#3427)

Browse Source
This commit is contained in:
George Hotz
2024-02-16 10:18:09 +01:00
committed by GitHub
parent 2d702ca073
commit fe97a85014
3 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -15,7 +15,7 @@ setup(name='tinygrad',
long_description=long_description,
long_description_content_type='text/markdown',
packages = ['tinygrad', 'tinygrad.runtime.autogen', 'tinygrad.codegen', 'tinygrad.nn', 'tinygrad.renderer',
'tinygrad.runtime', 'tinygrad.runtime.compiler', 'tinygrad.runtime.graph', 'tinygrad.shape', 'tinygrad.features'],
'tinygrad.runtime', 'tinygrad.runtime.driver', 'tinygrad.runtime.graph', 'tinygrad.shape', 'tinygrad.features'],
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License"

0
tinygrad/runtime/compiler/hip_comgr.py → tinygrad/runtime/driver/hip_comgr.py
View File

3
tinygrad/runtime/ops_hip.py
View File

@@ -7,8 +7,7 @@ from tinygrad.helpers import from_mv, round_up, to_mv, colored, init_c_struct_t
from tinygrad.device import Compiled, LRUAllocator, BufferOptions, JITRunner, Device, Buffer, MallocAllocator, update_stats, Compiler
from tinygrad.renderer.cstyle import HIPRenderer
from tinygrad.codegen.kernel import LinearizerOptions
from tinygrad.runtime.compiler.hip_comgr import compile_hip
from tinygrad.runtime.driver.hip_comgr import compile_hip
class HIPCompiler(Compiler):
linearizer_opts = LinearizerOptions("HIP", has_tensor_cores=True)