the compiler is a driver (tinygrad#3427)

ROCm · Feb 16, 2024 · fe97a85 · fe97a85
1 parent 2d702ca
commit fe97a85
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@
       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"

diff --git a/tinygrad/runtime/compiler/hip_comgr.py → tinygrad/runtime/driver/hip_comgr.py b/tinygrad/runtime/compiler/hip_comgr.py → tinygrad/runtime/driver/hip_comgr.py
diff --git a/tinygrad/runtime/ops_hip.py b/tinygrad/runtime/ops_hip.py
@@ -7,8 +7,7 @@
 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)