MCPcopy Create free account
hub / github.com/OpenDriveLab/ReSim / scaled_mask_softmax.py

File scaled_mask_softmax.py

SwissArmyTransformer/sat/ops/scaled_mask_softmax.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1try:
2 from apex.transformer.functional import FusedScaleMaskSoftmax
3 from apex.transformer.enums import AttnMaskType
4except ModuleNotFoundError:

Callers

nothing calls this directly

Calls 1

print_rank0Function · 0.90

Tested by

no test coverage detected