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

File multi.py

tinygrad/schedule/multi.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1from tinygrad.helpers import all_same, prod, getenv, ALLREDUCE_CAST
2from tinygrad.uop.ops import Ops, UOp, PatternMatcher, UPat, GroupOp, graph_rewrite
3from tinygrad.dtype import dtypes
4from tinygrad.schedule.allreduce import handle_allreduce

Callers

nothing calls this directly

Calls 11

PatternMatcherClass · 0.90
UPatClass · 0.90
UOpClass · 0.90
getenvFunction · 0.90
copy_to_deviceMethod · 0.80
mselectMethod · 0.80
multiMethod · 0.80
allreduceMethod · 0.80
gettupleMethod · 0.80
varMethod · 0.45
replaceMethod · 0.45

Tested by

no test coverage detected