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
Method__init__
(self, is_backward: bool, na_dim: int, sm: int)
scripts/autogen_fna.py:407
Method__init__
( self, is_backward: bool, na_dim: int, sm: int, dtype: DataType,
scripts/autogen_fna.py:458
Method__init__
(self, name, short_name, torch_name, bits)
scripts/autogen_blackwell_fmha_bwd.py:191
Method__init__
( self, dtype: DataType, gemm_shape: Tuple[int, int, int], )
scripts/autogen_blackwell_fmha_bwd.py:208
Method__init__
(self)
scripts/autogen_blackwell_fmha_bwd.py:291
Method__init__
(self, dtype: DataType)
scripts/autogen_blackwell_fmha_bwd.py:326
Method__init__
( self, dtype: DataType, head_dim: int, )
scripts/autogen_blackwell_fmha_bwd.py:365
Method__init__
(self, name, short_name, torch_name, bits)
scripts/autogen_blackwell_fna.py:119
Method__init__
( self, na_dim: int, dtype: DataType, gemm_shape: Tuple[int, int, int],
scripts/autogen_blackwell_fna.py:143
Method__init__
(self)
scripts/autogen_blackwell_fna.py:262
Method__init__
(self, na_dim: int)
scripts/autogen_blackwell_fna.py:295
Method__init__
(self, na_dim: int, dtype: DataType)
scripts/autogen_blackwell_fna.py:330
Method__init__
( self, na_dim: int, dtype: DataType, head_dim: int, )
scripts/autogen_blackwell_fna.py:370
Method__init__
( self, na_dim: int, dtype: DataType, head_dim: int, causal_mask: tupl
scripts/autogen_blackwell_fna.py:428
Method__init__
(self, name, short_name, torch_name, bits)
scripts/autogen_blackwell_fna_bwd.py:130
Method__init__
( self, na_dim: int, dtype: DataType, gemm_shape: Tuple[int, int, int],
scripts/autogen_blackwell_fna_bwd.py:152
Method__init__
(self)
scripts/autogen_blackwell_fna_bwd.py:266
Method__init__
(self, na_dim: int)
scripts/autogen_blackwell_fna_bwd.py:299
Method__init__
(self, na_dim: int, dtype: DataType)
scripts/autogen_blackwell_fna_bwd.py:334
Method__init__
( self, na_dim: int, dtype: DataType, head_dim: int, )
scripts/autogen_blackwell_fna_bwd.py:374
Method__init__
( self, na_dim: int, dtype: DataType, head_dim: int, causal_mask: tupl
scripts/autogen_blackwell_fna_bwd.py:434
Method__init__
(self, name, short_name, torch_name, bits)
scripts/autogen_reference_fna.py:172
Method__init__
( self, na_dim: int, dtype: DataType, causal: tuple, is_backward: bool
scripts/autogen_reference_fna.py:195
Method__init__
(self, is_backward: bool)
scripts/autogen_reference_fna.py:292
Method__init__
(self, is_backward: bool, na_dim: int)
scripts/autogen_reference_fna.py:329
Method__init__
( self, is_backward: bool, na_dim: int, dtype: DataType, )
scripts/autogen_reference_fna.py:367
Method__init__
(self, name, short_name, torch_name, bits)
scripts/autogen_blackwell_fmha.py:156
Method__init__
( self, dtype: DataType, gemm_shape: Tuple[int, int, int], is_persistent: bool
scripts/autogen_blackwell_fmha.py:175
Method__init__
(self)
scripts/autogen_blackwell_fmha.py:263
Method__init__
(self, dtype: DataType)
scripts/autogen_blackwell_fmha.py:296
Method__init__
( self, dtype: DataType, head_dim: int, )
scripts/autogen_blackwell_fmha.py:335
Method__init__
(self, name, short_name, torch_name, bits)
scripts/autogen_hopper_fmha_bwd.py:173
Method__init__
( self, dtype: DataType, gemm_shape: Tuple[int, int, int], )
scripts/autogen_hopper_fmha_bwd.py:190
Method__init__
(self)
scripts/autogen_hopper_fmha_bwd.py:273
Method__init__
(self, dtype: DataType)
scripts/autogen_hopper_fmha_bwd.py:308
Method__init__
( self, dtype: DataType, head_dim: int, )
scripts/autogen_hopper_fmha_bwd.py:347
Method__init__
(self, name, short_name, torch_name, bits)
scripts/autogen_hopper_fmha.py:182
Method__init__
( self, dtype: DataType, gemm_shape: Tuple[int, int, int], kernel_type: Kernel
scripts/autogen_hopper_fmha.py:222
Method__init__
(self)
scripts/autogen_hopper_fmha.py:309
Method__init__
(self, dtype: DataType)
scripts/autogen_hopper_fmha.py:342
Method__init__
( self, dtype: DataType, head_dim: int, )
scripts/autogen_hopper_fmha.py:379
Method__init__
(self, name, short_name, torch_name, bits)
scripts/autogen_hopper_fna.py:141
Method__init__
( self, na_dim: int, dtype: DataType, gemm_shape: Tuple[int, int, int],
scripts/autogen_hopper_fna.py:186
Method__init__
( self, )
scripts/autogen_hopper_fna.py:304
Method__init__
(self, na_dim: int)
scripts/autogen_hopper_fna.py:339
Method__init__
(self, na_dim: int, dtype: DataType)
scripts/autogen_hopper_fna.py:374
Method__init__
( self, na_dim: int, dtype: DataType, head_dim: int, )
scripts/autogen_hopper_fna.py:414
Method__init__
( self, na_dim: int, dtype: DataType, head_dim: int, causal_mask: tupl
scripts/autogen_hopper_fna.py:472
Method__init__
(self, sm: int, gemm_shape: Tuple[int, int, int])
scripts/autogen_fmha.py:34
Method__init__
(self, sm: int, gemm_shapes: List[Tuple[int, int, int]])
scripts/autogen_fmha.py:46
Method__init__
(self, name, torch_name, short_name, bits, min_sm)
scripts/autogen_fmha.py:64
Method__init__
( self, is_backward: bool, dtype: DataType, config: KernelConfig, )
scripts/autogen_fmha.py:114
Method__init__
( self, is_backward: bool, sm: int, dtype: DataType, config_list: Kern
scripts/autogen_fmha.py:176
Method__init__
(self, is_backward: bool)
scripts/autogen_fmha.py:272
Method__init__
(self, is_backward: bool, sm: int)
scripts/autogen_fmha.py:318
Method__init__
(self, path)
scripts/testing/pytest_progress_plugin.py:23
Method__init__
( self, batch: int, heads: int, head_dim: int, input_shape: DimensionT
tests/utils.py:52
Method__init__
( self, embed_dim: int, num_heads: int, qkv_bias: bool = True, qk_scal
tests/test_torch_compile.py:49
Method__init__
( self, embed_dim: int, num_heads: int, mlp_ratio: int, qkv_bias: bool
tests/test_torch_compile.py:118
Method__le__
(self, other)
src/natten/profiling_utils/formatting.py:169
Method__lt__
(self, other)
src/natten/profiling_utils/formatting.py:175
Method__ne__
(self, other)
src/natten/profiling_utils/formatting.py:184
Method__radd__
(self, other)
src/natten/profiling_utils/formatting.py:187
Method__repr__
(self)
src/natten/profiling_utils/problem.py:224
Function_attention_fn
( q, k, v, cumseq_q, cumseq_kv, max_q, max_kv, backend, is_causal=False )
tests/test_varlen_recompile.py:106
Function_get_possible_kv_splits
( min_splits: DimensionType, max_splits: DimensionType, )
src/natten/backends/configs/cutlass/backward_knobs.py:104
Function_handle_term
(*_)
scripts/testing/monitor.py:29
Function_maybe_torch_compile
(*args, **kwargs)
src/natten/attn_merge.py:38
Function_prologue
csrc/include/natten/cuda/fmha/gemm/custom_mma_multistage.h:375
Function_prologue
csrc/include/natten/cuda/fna/gemm/custom_mma_multistage.h:375
MethodaccumApplyExpToSmem
csrc/include/natten/cuda/fmha/gemm/mma_from_smem.h:1598
MethodaccumApplyExpToSmem
csrc/include/natten/cuda/fmha/gemm/mma_from_smem.h:1755
MethodaccumApplyExpToSmem
csrc/include/natten/cuda/fmha/gemm/mma_from_smem.h:1884
MethodaccumApplyExpToSmem
csrc/include/natten/cuda/fna/gemm/mma_from_smem.h:1608
MethodaccumApplyExpToSmem
csrc/include/natten/cuda/fna/gemm/mma_from_smem.h:1907
MethodaccumToSmem
csrc/include/natten/cuda/fmha/gemm/mma_from_smem.h:1578
MethodaccumToSmem
csrc/include/natten/cuda/fmha/gemm/mma_from_smem.h:1669
MethodaccumToSmem
csrc/include/natten/cuda/fmha/gemm/mma_from_smem.h:1828
MethodaccumToSmem
csrc/include/natten/cuda/fna/gemm/mma_from_smem.h:1588
MethodaccumToSmem
csrc/include/natten/cuda/fna/gemm/mma_from_smem.h:1851
FunctionaccumulateInGmem
csrc/include/natten/cuda/fna/kernel_backward.h:2608
MethodaccumulateInGmem
csrc/include/natten/cuda/fmha/kernel_backward.h:2134
Methodacquire
csrc/include/natten/cuda/fmha/kernel_backward.h:151
Methodacquire
csrc/include/natten/cuda/fna/kernel_backward.h:181
Functionadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/epilogue_predicated_tile_iterator.h:284
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:593
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1368
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1587
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1815
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:2042
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:309
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:592
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1061
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1339
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1566
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1808
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:2037
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator_residual_last.h:898
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:1361
Methodadd_pointer_offset
Adds a pointer offset in units of Element
csrc/include/natten/cuda/fna/iterators/predicated_tile_iterator.h:221
← previousnext →901–1,000 of 2,059, ranked by callers