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

create engine folder and move code (#3948)

Browse Source 
* retry

* older tf

* that
This commit is contained in:
George Hotz
2024-03-26 20:38:03 -07:00
committed by GitHub
parent 629cbc5587
commit 150ea2eb76
51 changed files with 54 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/runtime/graph/cuda.py
View File

@@ -5,7 +5,7 @@ from tinygrad.helpers import init_c_var, GraphException
from tinygrad.device import CompiledASTRunner, update_stats, Buffer, MultiDeviceJITGraph, BufferXfer
from tinygrad.runtime.ops_cuda import CUDADevice, check, encode_args, cu_time_execution
from tinygrad.shape.symbolic import Variable
from tinygrad.features.jit import JitItem, get_input_replace, get_jit_stats, \
from tinygrad.engine.jit import JitItem, get_input_replace, get_jit_stats, \
get_jc_idxs_with_updatable_launch_dims, get_jc_idxs_with_updatable_var_vals
class CUDAGraph(MultiDeviceJITGraph):