MCPcopy Create free account

hub / github.com/SHI-Labs/NATTEN / functions

Functions2,059 in github.com/SHI-Labs/NATTEN

⨍ Functions 2,059◇ Types & classes 573
Methodcheck_supported
csrc/include/natten/cuda/fmha/kernel_forward.h:514
Methodcheck_supported
csrc/include/natten/cuda/fmha/kernel_backward.h:1092
Methodcheck_supported
csrc/include/natten/cuda/fna/kernel_forward.h:558
Methodcheck_supported
csrc/include/natten/cuda/fna/kernel_backward.h:1204
Functionclean_env
()
tests/test_log.py:11
Functionclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:412
Functionclear_mask
< Efficiently disables all accesses guarded by mask
csrc/include/natten/cuda/fmha/iterators/epilogue_predicated_tile_iterator.h:727
Functionclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator_residual_last.h:410
Functionclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:857
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:637
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1413
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1632
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1859
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:2086
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:344
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:623
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1096
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1370
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1597
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1839
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:2068
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:1405
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:1628
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_iterator.h:256
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_iterator.h:534
Methodclear_mask
< Efficiently disables all accesses guarded by mask
csrc/include/natten/cuda/fna/iterators/epilogue_predicated_tile_iterator.h:528
Methodclear_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_iterator_residual_last.h:267
Methodcompute
csrc/include/natten/cuda/fmha_blackwell/kernel/sm100_fmha_bwd_kernel_tma_warpspecialized.hpp:1383
Methodcompute
csrc/include/natten/cuda/fna_hopper/collective/fna_collective_tma_warpspecialized.hpp:564
Methodcompute
csrc/include/natten/cuda/fna_hopper/collective/fna_collective_bwd_tma_warpspecialized.hpp:1032
Methodcompute
csrc/include/natten/cuda/fna_hopper/collective/fna_collective_tma.hpp:325
Methodcompute
csrc/include/natten/cuda/fna_blackwell/kernel/sm100_fna_bwd_kernel_tma_warpspecialized.hpp:1421
Methodcompute
csrc/include/natten/cuda/fmha_hopper/collective/fmha_collective_tma_warpspecialized.hpp:494
Methodcompute
csrc/include/natten/cuda/fmha_hopper/collective/fmha_collective_bwd_tma_warpspecialized.hpp:862
FunctioncomputeDelta
csrc/include/natten/cuda/fna/kernel_backward.h:2654
MethodcomputeDelta
csrc/include/natten/cuda/fmha/kernel_backward.h:2178
Methodcompute_expected_turn
csrc/include/natten/cuda/fmha_blackwell/kernel/sm100_fmha_bwd_kernel_tma_warpspecialized.hpp:442
Methodcompute_predicates_
Computes predicates based on internally tracked per-thread offset.
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1038
Methodcompute_predicates_
Computes predicates based on internally tracked per-thread offset.
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:620
Methodcompute_predicates_
Computes predicates based on internally tracked per-thread offset.
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:994
Methodcompute_source_needed_
Streams the result to global memory
csrc/include/natten/cuda/fmha/epilogue/epilogue_pipelined.h:446
Methodcompute_source_needed_
Streams the result to global memory
csrc/include/natten/cuda/fna/epilogue/epilogue_pipelined.h:446
Methodcompute_source_not_needed_
Streams the result to global memory
csrc/include/natten/cuda/fmha/epilogue/epilogue_pipelined.h:324
Methodcompute_source_not_needed_
Streams the result to global memory
csrc/include/natten/cuda/fna/epilogue/epilogue_pipelined.h:324
Methodconfigs
(self)
scripts/autogen_fna.py:56
Methodconfigs
(self)
scripts/autogen_fmha.py:52
Methodcopy
csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_kernel_bwd_convert.hpp:125
Methodcorrect_qkv_shape
csrc/include/natten/cuda/fna_hopper/collective/fna_fusion_bwd.hpp:241
Methodcorrect_qkv_shape
csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion_bwd.hpp:239
Methodcorrection_epilogue
csrc/include/natten/cuda/fmha_blackwell/collective/sm100_fmha_fwd_mainloop_tma_warpspecialized.hpp:878
Methodcorrection_epilogue
csrc/include/natten/cuda/fna_blackwell/collective/sm100_fna_fwd_mainloop_tma_warpspecialized.hpp:959
Methodcorrection_rescale
csrc/include/natten/cuda/fmha_blackwell/collective/sm100_fmha_fwd_mainloop_tma_warpspecialized.hpp:976
Methodcorrection_rescale
csrc/include/natten/cuda/fna_blackwell/collective/sm100_fna_fwd_mainloop_tma_warpspecialized.hpp:1057
Methodcpp_class
(self)
scripts/autogen_fna.py:149
Methodcpp_class
(self)
scripts/autogen_fmha.py:132
Functiondecorator
(f)
src/natten/attn_merge.py:39
Functiondecorator
(f)
src/natten/utils/testing.py:32
Functiondisable_flex_compile
Disallow Flex Attention + `torch.compile` entirely.
src/natten/context.py:224
Functiondisable_flex_compile_backprop
Disallow Flex Attention + `torch.compile` for backpropagation entirely.
src/natten/context.py:229
Functiondisabled_register_fake
(op, fn=None, /, *, lib=None, _stacklevel=1)
src/natten/_libnatten/torch_wrappers.py:89
Functiondisabled_register_op
( name, fn=None, /, *, mutates_args, device_types=None, schema=None )
src/natten/_libnatten/torch_wrappers.py:78
Functiondownsample_load_with_byte_offset
Loads a fragment from memory
csrc/include/natten/cuda/fmha/iterators/epilogue_predicated_tile_iterator.h:503
Functiondry_run
( problem: Problem, backend: Optional[str], fmha_backend: Optional[str], backprop: bool, t
src/natten/profiling_utils/dry_run.py:265
Functionenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:418
Functionenable_mask
< Efficiently enables all accesses guarded by mask
csrc/include/natten/cuda/fmha/iterators/epilogue_predicated_tile_iterator.h:732
Functionenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator_residual_last.h:416
Functionenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:863
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:643
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1419
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1638
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1865
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:2092
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:355
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:634
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1107
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1381
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1608
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1850
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:2079
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:1411
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:1634
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_iterator.h:262
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_iterator.h:540
Methodenable_mask
< Efficiently enables all accesses guarded by mask
csrc/include/natten/cuda/fna/iterators/epilogue_predicated_tile_iterator.h:533
Methodenable_mask
Clears the predicate set efficiently
csrc/include/natten/cuda/fna/iterators/predicated_tile_iterator_residual_last.h:278
Methodenable_mask
< Efficiently enables all accesses guarded by mask
csrc/include/natten/cuda/fna/epilogue/predicated_tile_iterator.h:609
Methodepilogue
csrc/include/natten/cuda/fmha_blackwell/kernel/sm100_fmha_bwd_kernel_tma_warpspecialized.hpp:1257
Methodepilogue
csrc/include/natten/cuda/fna_blackwell/kernel/sm100_fna_bwd_kernel_tma_warpspecialized.hpp:1290
Methodepilogue
csrc/include/natten/cuda/fmha_hopper/kernel/fmha_kernel_tma_warpspecialized.hpp:472
Methodepilogue
csrc/include/natten/cuda/fmha_hopper/kernel/fmha_kernel_tma.hpp:242
Methodepilogue_clear
csrc/include/natten/cuda/fmha_blackwell/kernel/sm100_fmha_bwd_kernel_tma_warpspecialized.hpp:1207
Methodepilogue_clear
csrc/include/natten/cuda/fna_blackwell/kernel/sm100_fna_bwd_kernel_tma_warpspecialized.hpp:1240
Functionextent_column
Extent of the matrix in columns
csrc/include/natten/cuda/fmha/iterators/epilogue_predicated_tile_iterator.h:683
Methodextent_column
Extent of the matrix in columns
csrc/include/natten/cuda/fna/iterators/epilogue_predicated_tile_iterator.h:484
Methodextent_column
Extent of the matrix in columns
csrc/include/natten/cuda/fna/epilogue/predicated_tile_iterator.h:507
Functionextent_row
Extent of the matrix in rows
csrc/include/natten/cuda/fmha/iterators/epilogue_predicated_tile_iterator.h:677
Methodextra_repr
(self)
src/natten/modules.py:124
Functionflatten
csrc/include/natten/natten.h:62
Functionfmha_backward_torch_fake_op
( query: Tensor, key: Tensor, value: Tensor, output: Tensor, d_output: Tensor, logsume
src/natten/_libnatten/torch_wrappers.py:622
Functionfmha_backward_torch_op
( query: Tensor, key: Tensor, value: Tensor, output: Tensor, d_output: Tensor, logsume
src/natten/_libnatten/torch_wrappers.py:539
← previousnext →1,101–1,200 of 2,059, ranked by callers