2 Commits

Author	SHA1	Message	Date
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
Philippe Tillet	53fd9631ef	[PYTHON] Added automated benchmark script (#63) ... This adds a bench functionality to the setup.py that can be used to run the benchmark suite and generates a bunch of csv files (and optionally plots) python setup.py bench python setup.py bench --with-plots python setup.py bench --filter=cross_entropy	2021-02-08 15:16:41 -05:00