MCPcopy Create free account

hub / github.com/facebookresearch/xformers / functions

Functions1,445 in github.com/facebookresearch/xformers

⨍ Functions 1,445◇ Types & classes 346↳ Endpoints 17
Functionstore_v
xformers/csrc/attention/hip_decoder/ck_tile_attention_forward_decoder_splitk.h:49
FunctionsumN
(output_ptr, scaling_ptr, *inputs, BLOCK_SIZE: tl.constexpr)
tests/test_triton_varargs.py:43
Methodsupports
(self, op: "SwiGLUOpDispatch")
xformers/ops/swiglu_op.py:94
Methodsupports
(cls, d: Inputs)
xformers/ops/fmha/common.py:416
Functionswin_attention_pattern
(H, W, window_size, shift_size=0)
xformers/components/attention/attention_patterns.py:124
Functiontemp_files_ctx
A context to get tempfiles and ensure they are cleaned up.
xformers/benchmarks/utils.py:220
Functiontest_Linear_bias
()
stubs/torch_stub_tests.py:624
Functiontest_Linear_to
()
stubs/torch_stub_tests.py:606
Functiontest_Module_eval
()
stubs/torch_stub_tests.py:613
Functiontest_Module_train
()
stubs/torch_stub_tests.py:618
Functiontest_Sized
()
stubs/torch_stub_tests.py:1659
Functiontest_Tensor_init
()
stubs/torch_stub_tests.py:1767
Functiontest__lt__
()
stubs/torch_stub_tests.py:1687
Functiontest_abs
()
stubs/torch_stub_tests.py:1605
Functiontest_adaptive_average_pool2d
()
stubs/torch_stub_tests.py:779
Functiontest_add
()
stubs/torch_stub_tests.py:875
Functiontest_alibi_pattern
()
tests/test_attention_patterns.py:209
Functiontest_all
()
stubs/torch_stub_tests.py:512
Functiontest_allclose
()
stubs/torch_stub_tests.py:715
Functiontest_analyze_prof
(dtype)
tests/test_profiler.py:157
Functiontest_analyze_prof_memeff
(op, causal: bool)
tests/test_profiler.py:209
Functiontest_analyze_prof_sdpa
(dtype, backend, causal: bool)
tests/test_profiler.py:179
Functiontest_and
()
stubs/torch_stub_tests.py:940
Functiontest_arange
(n: N)
stubs/torch_stub_tests.py:262
Functiontest_argmax
()
stubs/torch_stub_tests.py:1227
Functiontest_argmin
()
stubs/torch_stub_tests.py:1258
Functiontest_argsort
()
stubs/torch_stub_tests.py:684
Functiontest_attn_bias_blockdiag
()
tests/test_mem_eff_attention.py:1450
Functiontest_attn_bias_blockdiag_batched
()
tests/test_mem_eff_attention.py:1472
Functiontest_attn_bias_blockdiag_crossattn_causal
()
tests/test_mem_eff_attention.py:1496
Functiontest_attn_bias_blockdiag_crossattn_causal_with_prefix
()
tests/test_mem_eff_attention.py:1553
Functiontest_attn_bias_blockdiag_crossattn_causal_with_prefix_qk_cond
()
tests/test_mem_eff_attention.py:1539
Functiontest_attn_bias_blockdiag_doc
IMPORTANT: This is the example in the doc for `BlockDiagonalMask`. If this example needs to be updated, please also update the doc
tests/test_mem_eff_attention.py:1862
Functiontest_attn_bias_causal
()
tests/test_mem_eff_attention.py:1425
Functiontest_attn_bias_from_seqlens
()
tests/test_mem_eff_attention.py:1854
Functiontest_attn_bias_padded
()
tests/test_mem_eff_attention.py:1589
Functiontest_attn_bias_torch_tensor
()
tests/test_mem_eff_attention.py:1440
Functiontest_autograd_backward
()
stubs/torch_stub_tests.py:1647
Functiontest_backward
( opBW_device_dtype_biasT_B_Mq_Mkv_H_K_Kv, grad_out_contiguous, fmt, )
tests/test_mem_eff_attention.py:749
Functiontest_backward_gqa
(opBW)
tests/test_mem_eff_attention.py:2074
Functiontest_bitwise_not
()
stubs/torch_stub_tests.py:1057
Functiontest_bmm
(tensor_type, device)
tests/test_sparse_tensors.py:123
Functiontest_bmm
()
stubs/torch_stub_tests.py:833
Functiontest_bool
()
stubs/torch_stub_tests.py:1826
Functiontest_cat
()
stubs/torch_stub_tests.py:1334
Functiontest_cdist
()
stubs/torch_stub_tests.py:1071
Functiontest_checkpoint
(policy_fn, input_requires_grad, device, autocast)
tests/test_checkpoint.py:44
Functiontest_checkpoint_attention
(policy_fn, input_requires_grad, device, autocast, op)
tests/test_checkpoint.py:122
Functiontest_checkpoint_with_grad
(policy_fn, input_requires_grad, grad_mode)
tests/test_checkpoint.py:80
Functiontest_chunk
()
stubs/torch_stub_tests.py:1586
Functiontest_ck_splitk_decoder
( op, kv_heads: Optional[int], n_heads: int, padding: int, bsz: int, dtype: str, d
tests/test_mem_eff_attention.py:1797
Functiontest_clip_grad_norm
()
stubs/torch_stub_tests.py:1040
Functiontest_clip_grad_value
()
stubs/torch_stub_tests.py:1052
Functiontest_clone
()
stubs/torch_stub_tests.py:1090
Functiontest_compile_unflatten
()
tests/test_sparsity24.py:1071
Functiontest_consistency
( adjacents: bool, dim: int, padding: int, groups: int, internal_dtype: str, dtype_str
tests/test_rope_padded.py:166
Functiontest_constant_
()
stubs/torch_stub_tests.py:1729
Functiontest_contiguous
()
stubs/torch_stub_tests.py:663
Functiontest_conv2d
()
stubs/torch_stub_tests.py:1474
Functiontest_count_nonzero
()
stubs/torch_stub_tests.py:1318
Functiontest_cuda_streams
( opFW_device_dtype_biasT_B_Mq_Mkv_H_K_Kv, )
tests/test_mem_eff_attention.py:1152
Functiontest_cumsum
()
stubs/torch_stub_tests.py:647
Functiontest_cusparselt_format
(M: int, N: int)
tests/test_sparsity24.py:560
Functiontest_custom_scale
(opBW_device_dtype_biasT_B_Mq_Mkv_H_K_Kv)
tests/test_mem_eff_attention.py:1219
Functiontest_deepcopy
(tensor_type, device)
tests/test_sparse_tensors.py:218
Functiontest_detach2
()
tests/test_sparsity24.py:174
Functiontest_detach_requires_grad
()
tests/test_sparsity24.py:156
Functiontest_diag
()
stubs/torch_stub_tests.py:1389
Functiontest_diag_embed
()
stubs/torch_stub_tests.py:1117
Functiontest_diagonal
()
stubs/torch_stub_tests.py:1384
Functiontest_diff
()
stubs/torch_stub_tests.py:671
Functiontest_dilated_2d_pattern
(H, W, k)
tests/test_attention_patterns.py:156
Functiontest_dim
()
stubs/torch_stub_tests.py:1628
Functiontest_dispatch_decoding_bmghk
()
tests/test_mem_eff_attention.py:2211
Functiontest_dispatch_decoding_bmhk
()
tests/test_mem_eff_attention.py:2188
Functiontest_division
()
stubs/torch_stub_tests.py:220
Functiontest_dropout_backward_ck
(q_len, kv_len, batch_size, k, p)
tests/test_mem_eff_attention.py:1052
Functiontest_dropout_ck
(q_len, kv_len, batch_size, k_len, p, seed, attn_bias)
tests/test_mem_eff_attention.py:994
Functiontest_einsum
()
stubs/torch_stub_tests.py:1453
Functiontest_embedding
()
stubs/torch_stub_tests.py:284
Functiontest_empty
()
stubs/torch_stub_tests.py:424
Functiontest_empty_like
()
stubs/torch_stub_tests.py:460
Functiontest_empty_tensors_empty_b
( opFW_device_dtype_biasT_B_Mq_Mkv_H_K_Kv, )
tests/test_mem_eff_attention.py:2367
Functiontest_empty_tensors_empty_kv
( opFW_device_dtype_biasT_B_Mq_Mkv_H_K_Kv, )
tests/test_mem_eff_attention.py:2340
Functiontest_empty_tensors_empty_query
( opFW_device_dtype_biasT_B_Mq_Mkv_H_K_Kv, )
tests/test_mem_eff_attention.py:2315
Functiontest_enable_grad
()
stubs/torch_stub_tests.py:1613
Functiontest_env_vars
()
tests/test_multiprocessing_utils.py:32
Functiontest_equal
()
stubs/torch_stub_tests.py:1101
Functiontest_exception_in_bw_pass
()
tests/test_fwbw_overlap.py:144
Functiontest_exception_in_first_bw_pass
()
tests/test_fwbw_overlap.py:164
Functiontest_expand
()
stubs/torch_stub_tests.py:575
Functiontest_eye
()
stubs/torch_stub_tests.py:769
Methodtest_f16_biasf32
(self)
tests/test_mem_eff_attention.py:1920
Methodtest_f32_biasf16
(self)
tests/test_mem_eff_attention.py:1929
Functiontest_fav3_kvsplit_attn
( dtype_init, deterministic, causal, B, nheads_q, seq_len_q, seq_len_kv, max_l
tests/test_mem_eff_attention.py:2994
Functiontest_fft_fft2
()
stubs/torch_stub_tests.py:1747
Functiontest_flash_attention_decoder
(name, case)
xformers/benchmarks/benchmark_attn_decoding.py:416
Functiontest_flash_gqa_wrong_strides
()
tests/test_mem_eff_attention.py:2147
Functiontest_flatten
()
stubs/torch_stub_tests.py:389
Functiontest_floor_division
()
stubs/torch_stub_tests.py:187
← previousnext →1,101–1,200 of 1,445, ranked by callers