diff --git a/examples/handcode_resnet50_opt.py b/examples/handcode_resnet50_opt.py index 1b4c4339d7..9ab3a0a803 100644 --- a/examples/handcode_resnet50_opt.py +++ b/examples/handcode_resnet50_opt.py @@ -7,8 +7,12 @@ from tinygrad.codegen.linearizer import Linearizer from tinygrad.features.search import time_linearizer, beam_search, bufs_from_lin from tinygrad.helpers import ansilen, DEBUG, getenv from tinygrad.shape.symbolic import sym_infer +from tinygrad.dtype import dtypes if __name__ == "__main__": + if getenv("HALF"): + dtypes.default_float = dtypes.half + mdl = ResNet50() seen = set()