Jared Kaplan 045ab5d62a [PYTHON] Add Blocksparse Attention Fwd/Bwd Test (#69) ...

Also includes small bugfix for block-sparse softmax

2021-07-27 12:38:49 -07:00

..

test_blocksparse.py

[PYTHON] Add Blocksparse Attention Fwd/Bwd Test (#69)

2021-07-27 12:38:49 -07:00

test_conv.py

[PYTHON] Added automated benchmark script (#63)

2021-07-27 12:38:48 -07:00

test_cross_entropy.py

[PYTHON] Added automated benchmark script (#63)

2021-07-27 12:38:48 -07:00

test_matmul.py

[CI] Added benchmarking to CI script (#65)

2021-07-27 12:38:49 -07:00