mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

don't fail when termcolor is not installed (#436)

Browse source
This commit is contained in:
cloud11665 2022-11-15 01:45:06 +01:00 • committed by GitHub
commit 4fb97b8de0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
test/test_speed_v_torch.py
View file

@ -11,9 +11,11 @@ from tinygrad.tensor import Tensor
from tinygrad.nn import Conv2d
try:
from termcolor import colored
except ImportError:
colored = lambda x, _: x
try:
from tinygrad.llops.ops_gpu import CL
except ImportError:
colored = None
CL = None
IN_CHANS = [int(x) for x in os.getenv("IN_CHANS", "4,16,64").split(",")]