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

update variable names around jit [pr] (#14049)

Browse Source 
lbs, st_vars_dtype_device and rawbuffers no more
This commit is contained in:
chenyu
2026-01-06 22:32:41 -05:00
committed by GitHub
parent 2833c5a54b
commit 87f4bc5446
6 changed files with 55 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/test_pkl_imagenet.py
View File

@@ -7,7 +7,7 @@ if __name__ == "__main__":
with open(fetch(sys.argv[1]), "rb") as f:
run_onnx_jit = pickle.load(f)
input_name = run_onnx_jit.captured.expected_names[0]
device = run_onnx_jit.captured.expected_st_vars_dtype_device[0][-1]
device = run_onnx_jit.captured.expected_input_info[0][-1]
print(f"input goes into {input_name=} on {device=}")
hit = 0
for i,(img,y) in enumerate(imagenet_dataloader(cnt=getenv("CNT", 100))):