Functions — github.com/SHI-Labs/NATTEN

CodeHub 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

Functionfmha_forward_torch_fake_op

( query: Tensor, key: Tensor, value: Tensor, is_causal: bool, scale: float, q_tile_siz

src/natten/_libnatten/torch_wrappers.py:511

Functionfmha_forward_torch_op

( query: Tensor, key: Tensor, value: Tensor, is_causal: bool, scale: float, q_tile_siz

src/natten/_libnatten/torch_wrappers.py:459

Functionfna_backward_torch_fake_op

( query: Tensor, key: Tensor, value: Tensor, output: Tensor, d_output:

src/natten/_libnatten/torch_wrappers.py:1170

Functionfna_backward_torch_op

( query: Tensor, key: Tensor, value: Tensor, output: Tensor, d_output:

src/natten/_libnatten/torch_wrappers.py:1102

Functionfna_forward_torch_fake_op

( query: Tensor, key: Tensor, value: Tensor, kernel_size: list[int], s

src/natten/_libnatten/torch_wrappers.py:1073

Functionfna_forward_torch_op

( query: Tensor, key: Tensor, value: Tensor, kernel_size: list[int], s

src/natten/_libnatten/torch_wrappers.py:1034

Functionfna_reference_backward

csrc/include/natten/cuda/reference/fna_reference_backward.hpp:610

Functionfna_reference_forward

csrc/include/natten/cuda/reference/fna_reference_forward.hpp:216

( ctx, *args, )

src/natten/attn_merge.py:134

(self, x: Tensor)

src/natten/modules.py:83

( ctx, tensor: Tensor, tile_shape: DimensionType, dilation: Di

src/natten/token_permute/cutlass_impl.py:91

( ctx, tensor: Tensor, token_layout: DimensionType, tile_shape

src/natten/token_permute/cutlass_impl.py:147

( ctx, query: Tensor, key: Tensor, value: Tensor, is_causal: bool,

src/natten/backends/hopper_fmha.py:55

( ctx, query: Tensor, key: Tensor, value: Tensor,

src/natten/backends/reference.py:84

( ctx, query: Tensor, key: Tensor, value: Tensor, is_causal: bool,

src/natten/backends/blackwell_fmha.py:54

( ctx, query: Tensor, key: Tensor, value: Tensor,

src/natten/backends/blackwell_fna.py:91

( ctx, query: Tensor, key: Tensor, value: Tensor,

src/natten/backends/fna.py:93

( ctx, query: Tensor, key: Tensor, value: Tensor, is_causal: bool,

src/natten/backends/fmha.py:54

( ctx, query: Tensor, key: Tensor, value: Tensor,

src/natten/backends/hopper_fna.py:92

( self, x: torch.Tensor, additional_context: Optional[torch.Tensor] = None, *a

tests/test_torch_compile.py:67

(self, x: torch.Tensor, c: torch.Tensor, *args, **kwargs)

tests/test_torch_compile.py:144

( self, x: torch.Tensor, additional_context: Optional[torch.Tensor] = None, *a

tests/test_torch_compile.py:183

Functionfully_block_sparse

csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion.hpp:532

Returns a pointer

csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:323

Returns a pointer

csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator_residual_last.h:340

Returns a pointer

csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:785

Returns a pointer

csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:606

Returns a pointer

csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1382

Returns a pointer

csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1601

Returns a pointer

csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1828

Returns a pointer

csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:2055

Returns a pointer

csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator_residual_last.h:911

Returns a pointer

csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:1374

MethodgetBlocksGrid

csrc/include/natten/cuda/fmha/kernel_forward.h:287

MethodgetBlocksGrid

csrc/include/natten/cuda/fmha/kernel_backward.h:779

MethodgetBlocksGrid

csrc/include/natten/cuda/fna/kernel_forward.h:325

MethodgetBlocksGrid

csrc/include/natten/cuda/fna/kernel_backward.h:878

MethodgetNumParallelBlocksForQuery

Returns how many kernel blocks will write to a given block in `grad_query` This is usually equal to the number of key splits, but can be different for

csrc/include/natten/cuda/fmha/kernel_backward.h:2001

MethodgetQueryEnd

csrc/include/natten/cuda/fmha/kernel_backward.h:1973

MethodgetQueryStart

Iteration order logic

csrc/include/natten/cuda/fmha/kernel_backward.h:1969

MethodgetQueryStartShift

csrc/include/natten/cuda/fmha/kernel_backward.h:1961

MethodgetRowOffset

csrc/include/natten/cuda/fmha/epilogue/epilogue_pipelined.h:599

MethodgetRowOffset

csrc/include/natten/cuda/fna/epilogue/epilogue_pipelined.h:599

MethodgetSmallestQueryForKey

csrc/include/natten/cuda/fmha/kernel_backward.h:1977

MethodgetThreadsGrid

csrc/include/natten/cuda/fmha/kernel_forward.h:294

MethodgetThreadsGrid

csrc/include/natten/cuda/fmha/kernel_backward.h:783

MethodgetThreadsGrid

csrc/include/natten/cuda/fna/kernel_forward.h:333

MethodgetThreadsGrid

csrc/include/natten/cuda/fna/kernel_backward.h:883

FunctiongetWarpsPerSmBw

csrc/include/natten/cuda/fmha/kernel_backward.h:179

FunctiongetWarpsPerSmBw

csrc/include/natten/cuda/fna/kernel_backward.h:209

FunctiongetWarpsPerSmFw

csrc/include/natten/cuda/fmha/kernel_forward.h:57

FunctiongetWarpsPerSmFw

csrc/include/natten/cuda/fna/kernel_forward.h:85

Methodget_block_coord

csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_causal_tile_scheduler.hpp:93

Methodget_block_coord

csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_causal_tile_scheduler.hpp:199

Methodget_block_coord

csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_tile_scheduler.hpp:156

Methodget_block_coord

csrc/include/natten/cuda/fmha_hopper/kernel/fmha_tile_scheduler.hpp:160

Methodget_block_coord

csrc/include/natten/cuda/fmha_hopper/kernel/fmha_tile_scheduler.hpp:212

Methodget_block_shape

csrc/include/natten/cuda/fmha_blackwell/kernel/sm100_fmha_fwd_kernel_tma_warpspecialized.hpp:259

Methodget_block_shape

csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_kernel_bwd_convert.hpp:113

Methodget_block_shape

csrc/include/natten/cuda/fmha_blackwell/kernel/sm100_fmha_bwd_kernel_tma_warpspecialized.hpp:2320

Methodget_block_shape

csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_kernel_bwd_sum_OdO.hpp:113

Methodget_block_shape

csrc/include/natten/cuda/fna_blackwell/kernel/sm100_fna_bwd_kernel_tma_warpspecialized.hpp:2375

Methodget_block_shape

csrc/include/natten/cuda/fna_blackwell/kernel/sm100_fna_fwd_kernel_tma_warpspecialized.hpp:216

Methodget_block_shape

csrc/include/natten/cuda/fmha_hopper/kernel/fmha_kernel_tma_warpspecialized.hpp:165

Methodget_block_shape

csrc/include/natten/cuda/fmha_hopper/kernel/fmha_kernel_bwd_convert.hpp:108

Methodget_block_shape

csrc/include/natten/cuda/fmha_hopper/kernel/fmha_kernel_tma.hpp:131

Methodget_block_shape

csrc/include/natten/cuda/fmha_hopper/kernel/fmha_kernel_bwd_sum_OdO.hpp:113

Methodget_block_shape

csrc/include/natten/cuda/reduction/fmha_kernel_bwd_sum_OdO.hpp:97

Functionget_bwd_win_end

csrc/include/natten/cuda/fna_hopper/collective/fna_fusion_bwd.hpp:65

Functionget_bwd_win_end

csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion_bwd.hpp:65

Functionget_bwd_win_start

csrc/include/natten/cuda/fna_hopper/collective/fna_fusion_bwd.hpp:47

Functionget_bwd_win_start

csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion_bwd.hpp:47

Functionget_bwd_window_end

csrc/include/natten/cuda/fna_hopper/collective/fna_fusion_bwd.hpp:162

Functionget_bwd_window_end

csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion_bwd.hpp:161

Functionget_bwd_window_start

csrc/include/natten/cuda/fna_hopper/collective/fna_fusion_bwd.hpp:94

Functionget_bwd_window_start

csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion_bwd.hpp:94

Methodget_dispatcher

scripts/autogen_hopper_fna_bwd.py:312

Methodget_dispatcher

scripts/autogen_hopper_fna_bwd.py:351

Methodget_dispatcher

scripts/autogen_hopper_fna_bwd.py:397

Methodget_dispatcher

scripts/autogen_hopper_fna_bwd.py:489

Methodget_dispatcher

scripts/autogen_fna.py:368

Methodget_dispatcher

scripts/autogen_fna.py:422

Methodget_dispatcher

scripts/autogen_fna.py:486

Methodget_dispatcher

scripts/autogen_blackwell_fmha_bwd.py:335

Methodget_dispatcher

scripts/autogen_blackwell_fmha_bwd.py:403

Methodget_dispatcher

scripts/autogen_blackwell_fna.py:303

Methodget_dispatcher

scripts/autogen_blackwell_fna.py:342

Methodget_dispatcher

scripts/autogen_blackwell_fna.py:388

Methodget_dispatcher

scripts/autogen_blackwell_fna.py:476

Methodget_dispatcher

scripts/autogen_blackwell_fna_bwd.py:307

Methodget_dispatcher

scripts/autogen_blackwell_fna_bwd.py:346

Methodget_dispatcher

scripts/autogen_blackwell_fna_bwd.py:392

Methodget_dispatcher

scripts/autogen_blackwell_fna_bwd.py:481

Methodget_dispatcher

scripts/autogen_reference_fna.py:339

Methodget_dispatcher

scripts/autogen_reference_fna.py:398

Methodget_dispatcher

scripts/autogen_blackwell_fmha.py:305

Methodget_dispatcher

scripts/autogen_blackwell_fmha.py:376

Methodget_dispatcher

scripts/autogen_hopper_fmha_bwd.py:317

Methodget_dispatcher

scripts/autogen_hopper_fmha_bwd.py:389

Methodget_dispatcher

scripts/autogen_hopper_fmha.py:351

← previous next →1,201–1,300 of 2,059, ranked by callers