mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-08 22:48:25 -05:00
@@ -1,5 +1,6 @@
|
|||||||
import os, sys, pickle, time, re
|
import os, sys, pickle, time, re
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
if "JIT_BATCH_SIZE" not in os.environ: os.environ["JIT_BATCH_SIZE"] = "0"
|
||||||
|
|
||||||
from tinygrad import fetch, Tensor, TinyJit, Context, GlobalCounters, Device, dtypes
|
from tinygrad import fetch, Tensor, TinyJit, Context, GlobalCounters, Device, dtypes
|
||||||
from tinygrad.helpers import DEBUG, getenv
|
from tinygrad.helpers import DEBUG, getenv
|
||||||
|
|||||||
Reference in New Issue
Block a user