MCPcopy
hub / github.com/tinygrad/tinygrad / flat_llama.py

File flat_llama.py

examples/mlperf/models/flat_llama.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import math, os
2if __name__ == "__main__":
3 os.environ["DEFAULT_FLOAT"] = "bfloat16"
4 os.environ["OPTIM_DTYPE"] = "bfloat16"

Callers

nothing calls this directly

Calls 15

getenvFunction · 0.90
round_upFunction · 0.90
coloredFunction · 0.90
TimingClass · 0.90
profile_markerFunction · 0.90
FlatTransformerClass · 0.85
_make_gradFunction · 0.85
fwd_bwdFunction · 0.85
reshapeMethod · 0.80
arangeMethod · 0.80
realizeMethod · 0.80
shard_Method · 0.80

Tested by

no test coverage detected