Path	Lines of Code
xformers/benchmarks/LRA/__init__.py	1
xformers/benchmarks/LRA/batch_fetch_results.py	81
xformers/benchmarks/LRA/batch_submit.py	43
xformers/benchmarks/LRA/code/__init__.py	1
xformers/benchmarks/LRA/code/dataset.py	28
xformers/benchmarks/LRA/code/model_wrapper.py	147
xformers/benchmarks/LRA/run_grid_search.py	116
xformers/benchmarks/LRA/run_tasks.py	467
xformers/benchmarks/LRA/run_with_submitit.py	110
xformers/benchmarks/LRA/setup/cifar10.py	35
xformers/benchmarks/LRA/setup/listops.py	31
xformers/benchmarks/LRA/setup/pathfinder.py	44
xformers/benchmarks/LRA/setup/retrieval.py	33
xformers/benchmarks/LRA/setup/text.py	32
xformers/benchmarks/__init__.py	1
xformers/benchmarks/benchmark_core.py	225
xformers/benchmarks/benchmark_encoder.py	314
xformers/benchmarks/benchmark_nystrom_utils.py	76
xformers/benchmarks/benchmark_pytorch_transformer.py	183
xformers/benchmarks/benchmark_revnet.py	61
xformers/benchmarks/benchmark_sddmm.py	85
xformers/benchmarks/benchmark_triton_blocksparse.py	96
xformers/benchmarks/benchmark_triton_dropout.py	86
xformers/benchmarks/benchmark_triton_fused_linear.py	110
xformers/benchmarks/benchmark_triton_layernorm.py	61
xformers/benchmarks/benchmark_triton_softmax.py	61
xformers/benchmarks/benchmark_triton_stride_sum.py	47
xformers/benchmarks/benchmark_vit_timm.py	311
xformers/benchmarks/utils.py	79