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

Also includes small bugfix for block-sparse softmax

2021-02-19 17:46:05 -05:00

..

test_blocksparse.py

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

2021-02-19 17:46:05 -05:00

test_conv.py

[PYTHON] Added automated benchmark script (#63)

2021-02-08 15:16:41 -05:00

test_cross_entropy.py

[PYTHON] Added automated benchmark script (#63)

2021-02-08 15:16:41 -05:00

test_matmul.py

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

2021-02-10 14:01:47 -05:00