tests/L0/run_transformer/test_fused_softmax.py · 1337e81eebe5341ffda02a77929f05168dd5813c · OpenDAS / apex

Time dimension shape check for fused scale mask softmax kernel (#1421) · 1337e81e

Sandeep Subramanian authored Jul 13, 2022



* Time dimension shape check for fused scale mask softmax kernel
Signed-off-by: MaximumEntropy <sandeep.subramanian.1@umontreal.ca>

* Add shape test
Signed-off-by: MaximumEntropy <sandeep.subramanian.1@umontreal.ca>

* Fix mask shape
Signed-off-by: MaximumEntropy <sandeep.subramanian.1@umontreal.ca>

1337e81e

test_fused_softmax.py 8.41 KB

Replace test_fused_softmax.py