MCPcopy Create free account

hub / github.com/facebookresearch/xformers / endpoints

Endpoints17 in github.com/facebookresearch/xformers

⨍ Functions 1,445◇ Types & classes 346↳ Endpoints 17
Route_fused_allgather_and_linear_custom_op
torch.library.custom_op( "xformers_python::_fused_allgather_and_linear_impl", mutates_args={"gathered_
xformers/ops/sequence_parallel_fused_ops.py:None
Route_fused_linear_and_reducescatter_custom_op
torch.library.custom_op( "xformers_python::_fused_linear_and_reducescatter_impl", mutates_args={"scatt
xformers/ops/sequence_parallel_fused_ops.py:None
Route_is_triton_available
compute_once
xformers/__init__.py:None
Routeget_python_lib
compute_once
xformers/__init__.py:None
Routemha_bwd_flops
register_flop_formula(torch.ops.xformers_flash3.flash_bwd, get_raw=True)
xformers/ops/fmha/flash3.py:None
Routemha_fwd_flops
register_flop_formula(torch.ops.xformers_flash3.flash_fwd, get_raw=True)
xformers/ops/fmha/flash3.py:None
Routetest_checkpoint
pytest.mark.parametrize("input_requires_grad", [True, False])
tests/test_checkpoint.py:None
Routetest_checkpoint_attention
pytest.mark.parametrize("input_requires_grad", [True, False])
tests/test_checkpoint.py:None
Routetest_checkpoint_with_grad
pytest.mark.parametrize("input_requires_grad", [True, False])
tests/test_checkpoint.py:None
Routetest_flash_attention_decoder
pytest.mark.parametrize( "name, case", [(name, case) for name in get_benchmark_names() for case in TES
xformers/benchmarks/benchmark_attn_decoding.py:None
Routetest_forward_gqa
pytest.mark.parametrize( "opFW_biasT", [ (op, biasT) for op in ALL_FW_OPS for
tests/test_mem_eff_attention.py:None
Routetest_merge_attentions_decoding
pytest.mark.parametrize( "stack_inputs", (False, True), ids=lambda x: "stack_inputs" if x else "" )
tests/test_fmha_merge_attentions.py:None
Routetest_merge_attentions_nobias
pytest.mark.parametrize( "stack_inputs", (False, True), ids=lambda x: "stack_inputs" if x else "" )
tests/test_fmha_merge_attentions.py:None
Routetest_optimal_checkpoint_policy
pytest.mark.parametrize("memory_budget", [0, 0.03, 0.05, 0.1, 0.3, 0.5, 0.8, 1.0])
tests/test_checkpoint.py:None
Routetest_selective_checkpoint_wrapper_compile
pytest.mark.parametrize("memory_budget", [0, 0.1, 0.3, 1.0])
tests/test_checkpoint.py:None
Routetest_sequence_parallel_fused
pytest.mark.parametrize( "dtype", [ # https://fb.workplace.com/groups/1075192433118967/posts/1
tests/test_sequence_parallel_fused_ops.py:None
Routetest_sparsify24_like_dense
pytest.mark.parametrize("input_rowmajor", [True, False])
tests/test_sparsity24.py:None