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

File fa.py

extra/thunder/cuda/fa.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import pathlib
2from tinygrad import Device, Tensor
3from tinygrad.helpers import Context
4from tinygrad.runtime.support.compiler_cuda import pretty_ptx, NVCCCompiler

Callers

nothing calls this directly

Calls 15

NVCCCompilerClass · 0.90
pretty_ptxFunction · 0.90
ContextClass · 0.90
splitMethod · 0.80
randnMethod · 0.80
realizeMethod · 0.80
ensure_allocatedMethod · 0.80
scaled_dot_product_attentionMethod · 0.80
floatMethod · 0.80
itemMethod · 0.80
meanMethod · 0.80
compileMethod · 0.45

Tested by

no test coverage detected