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

s/tinytqdm/tqdm (#5103)

Browse Source 
except in unit test where tqdm is imported
This commit is contained in:
chenyu
2024-06-22 14:18:26 -04:00
committed by GitHub
parent 9f875123b6
commit 8080298739
8 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extra/optimization/get_action_space.py
View File

@@ -2,7 +2,7 @@ import random
from extra.optimization.helpers import load_worlds, ast_str_to_lin
from tinygrad.engine.search import actions
from tinygrad.codegen.linearizer import Linearizer
from tinygrad.helpers import tinytqdm
from tinygrad.helpers import tqdm
tactions = set()
def test_rebuild(lin):
@@ -20,7 +20,7 @@ if __name__ == "__main__":
ast_strs = load_worlds(False, False, False)
random.shuffle(ast_strs)
ast_strs = ast_strs[:2000]
for ast_str in tinytqdm(ast_strs):
for ast_str in tqdm(ast_strs):
lin = ast_str_to_lin(ast_str)
#if not lin.apply_tensor_cores():
lin.hand_coded_optimizations()