Philippe Tillet db941161ed [PYTHON][EXAMPLES] Cleaned self-attention benchmarks

2020-01-22 18:09:00 -05:00

..

bench.py

[PYTHON][EXAMPLES] Cleaned self-attention benchmarks

2020-01-22 18:09:00 -05:00

optimized.py

[PYTHON][EXAMPLES] Added self-attention example using triton.ops.einsum

2020-01-21 16:45:04 -05:00

reference.py

[PYTHON][EXAMPLES] Added self-attention example using triton.ops.einsum

2020-01-21 16:45:04 -05:00