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
Methodget_dispatcher
(self)
scripts/autogen_hopper_fmha.py:424
Methodget_dispatcher
(self)
scripts/autogen_hopper_fna.py:347
Methodget_dispatcher
(self)
scripts/autogen_hopper_fna.py:386
Methodget_dispatcher
(self)
scripts/autogen_hopper_fna.py:432
Methodget_dispatcher
(self)
scripts/autogen_hopper_fna.py:525
Methodget_dispatcher
(self)
scripts/autogen_fmha.py:333
Methodget_extra_trip_count
csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion.hpp:352
Methodget_extra_trip_info
csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion.hpp:364
Functionget_from_tuple
csrc/include/natten/natten.h:100
Methodget_grid_shape
Computes the grid shape
csrc/include/natten/cuda/fmha_blackwell/device/fmha_sm100.hpp:103
Methodget_grid_shape
csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_causal_tile_scheduler.hpp:84
Methodget_grid_shape
csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_causal_tile_scheduler.hpp:189
Methodget_grid_shape
csrc/include/natten/cuda/fmha_blackwell/kernel/sm100_fmha_fwd_kernel_tma_warpspecialized.hpp:255
Methodget_grid_shape
csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_kernel_bwd_convert.hpp:101
Methodget_grid_shape
csrc/include/natten/cuda/fmha_blackwell/kernel/sm100_fmha_bwd_kernel_tma_warpspecialized.hpp:2325
Methodget_grid_shape
csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_kernel_bwd_sum_OdO.hpp:105
Methodget_grid_shape
csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_tile_scheduler.hpp:69
Methodget_grid_shape
csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_tile_scheduler.hpp:146
Methodget_grid_shape
Computes the grid shape
csrc/include/natten/cuda/utils/generic_cutlass_device.hpp:104
Methodget_grid_shape
Computes the grid shape
csrc/include/natten/cuda/fna_hopper/device/fna_sm90.hpp:103
Methodget_grid_shape
Computes the grid shape
csrc/include/natten/cuda/fna_blackwell/device/fna_sm100.hpp:103
Methodget_grid_shape
csrc/include/natten/cuda/fna_blackwell/kernel/sm100_fna_bwd_kernel_tma_warpspecialized.hpp:2380
Methodget_grid_shape
csrc/include/natten/cuda/fna_blackwell/kernel/sm100_fna_fwd_kernel_tma_warpspecialized.hpp:212
Methodget_grid_shape
Computes the grid shape
csrc/include/natten/cuda/fmha_hopper/device/fmha_sm90.hpp:103
Methodget_grid_shape
csrc/include/natten/cuda/fmha_hopper/kernel/fmha_kernel_tma_warpspecialized.hpp:161
Methodget_grid_shape
csrc/include/natten/cuda/fmha_hopper/kernel/fmha_kernel_bwd_convert.hpp:96
Methodget_grid_shape
csrc/include/natten/cuda/fmha_hopper/kernel/fmha_kernel_tma.hpp:127
Methodget_grid_shape
csrc/include/natten/cuda/fmha_hopper/kernel/fmha_kernel_bwd_sum_OdO.hpp:105
Methodget_grid_shape
csrc/include/natten/cuda/fmha_hopper/kernel/fmha_tile_scheduler.hpp:69
Methodget_grid_shape
csrc/include/natten/cuda/fmha_hopper/kernel/fmha_tile_scheduler.hpp:150
Methodget_grid_shape
csrc/include/natten/cuda/fmha_hopper/kernel/fmha_tile_scheduler.hpp:203
Methodget_grid_shape
csrc/include/natten/cuda/reduction/fmha_kernel_bwd_sum_OdO.hpp:89
Methodget_lane_offset
csrc/include/natten/cuda/fmha/gemm/mma_accum_lambda_iterator.h:37
Methodget_lane_offset
csrc/include/natten/cuda/fmha/gemm/mma_accum_lambda_iterator.h:117
Methodget_lane_offset
csrc/include/natten/cuda/fmha/gemm/mma_accum_lambda_iterator.h:263
Methodget_lane_offset
csrc/include/natten/cuda/fna/gemm/mma_accum_lambda_iterator.h:71
Methodget_lane_offset
csrc/include/natten/cuda/fna/gemm/mma_accum_lambda_iterator.h:154
Methodget_lane_offset
csrc/include/natten/cuda/fna/gemm/mma_accum_lambda_iterator.h:302
Functionget_logger
(name)
src/natten/utils/log.py:133
Functionget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:430
Functionget_mask
< Sets the mask
csrc/include/natten/cuda/fmha/iterators/epilogue_predicated_tile_iterator.h:737
Functionget_mask
Gets the mask
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator_residual_last.h:428
Functionget_mask
Gets the mask
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:875
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:655
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1431
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1650
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:1877
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_access_iterator_residual_last.h:2104
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:367
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:646
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1119
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1393
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1620
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:1862
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fmha/iterators/predicated_tile_iterator_residual_last.h:2091
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:1423
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fna/iterators/predicated_tile_access_iterator.h:1646
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fna/iterators/predicated_tile_iterator.h:274
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fna/iterators/predicated_tile_iterator.h:552
Methodget_mask
< Sets the mask
csrc/include/natten/cuda/fna/iterators/epilogue_predicated_tile_iterator.h:538
Methodget_mask
Gets the mask
csrc/include/natten/cuda/fna/iterators/predicated_tile_iterator_residual_last.h:290
Methodget_mask
< Sets the mask
csrc/include/natten/cuda/fna/epilogue/predicated_tile_iterator.h:614
Methodget_masked_trip_count
csrc/include/natten/cuda/fmha_blackwell/collective/fmha_fusion.hpp:79
Methodget_masked_trip_count
csrc/include/natten/cuda/fmha_blackwell/collective/fmha_fusion.hpp:125
Methodget_masked_trip_count
csrc/include/natten/cuda/fmha_blackwell/collective/fmha_fusion.hpp:199
Methodget_masked_trip_count
csrc/include/natten/cuda/fmha_hopper/collective/fmha_fusion.hpp:52
Methodget_masked_trip_count
csrc/include/natten/cuda/fmha_hopper/collective/fmha_fusion.hpp:82
Functionget_min_splits
(na_dim: int)
src/natten/backends/configs/cutlass/backward_knobs.py:56
Methodget_name
(self, is_backward: bool)
scripts/autogen_fna.py:62
Methodget_name
(self, is_backward: bool)
scripts/autogen_fmha.py:58
Methodget_problem_shape_O
csrc/include/natten/cuda/fmha_blackwell/collective/sm100_fmha_fwd_epilogue_tma_warpspecialized.hpp:100
Methodget_problem_shape_O
csrc/include/natten/cuda/fna_blackwell/collective/sm100_fna_fwd_epilogue_tma_warpspecialized.hpp:100
Methodget_trip_count
csrc/include/natten/cuda/fmha_blackwell/collective/fmha_fusion.hpp:177
Methodget_trip_count
csrc/include/natten/cuda/fna_hopper/collective/fna_fusion_bwd.hpp:264
Methodget_trip_count
csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion_bwd.hpp:261
Methodget_trip_count
csrc/include/natten/cuda/fmha_hopper/collective/fmha_fusion.hpp:44
Methodget_trip_count
csrc/include/natten/cuda/fmha_hopper/collective/fmha_fusion.hpp:177
Methodget_trip_count
csrc/include/natten/cuda/fmha_hopper/collective/fmha_fusion.hpp:238
Methodget_trip_info
csrc/include/natten/cuda/fna_hopper/collective/fna_collective_bwd_tma_warpspecialized.hpp:570
Methodget_unmasked_trip_count
csrc/include/natten/cuda/fmha_blackwell/collective/fmha_fusion.hpp:90
Methodget_unmasked_trip_count
csrc/include/natten/cuda/fmha_blackwell/collective/fmha_fusion.hpp:136
Methodget_unmasked_trip_count
csrc/include/natten/cuda/fmha_blackwell/collective/fmha_fusion.hpp:218
Methodget_unmasked_trip_count
csrc/include/natten/cuda/fmha_hopper/collective/fmha_fusion.hpp:60
Methodget_unmasked_trip_count
csrc/include/natten/cuda/fmha_hopper/collective/fmha_fusion.hpp:90
Functionget_win_end
csrc/include/natten/cuda/reference/mask.hpp:83
Functionget_win_end
csrc/include/natten/cuda/fna_hopper/collective/fna_fusion.hpp:68
Functionget_win_end
csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion.hpp:114
Functionget_win_start
csrc/include/natten/cuda/reference/mask.hpp:60
Functionget_win_start
csrc/include/natten/cuda/fna_hopper/collective/fna_fusion.hpp:45
Functionget_win_start
csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion.hpp:89
Functionget_window_end
csrc/include/natten/cuda/reference/mask.hpp:145
Functionget_window_end
csrc/include/natten/cuda/fna_hopper/collective/fna_fusion.hpp:135
Functionget_window_end
csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion.hpp:181
Functionget_window_start
csrc/include/natten/cuda/reference/mask.hpp:92
Functionget_window_start
csrc/include/natten/cuda/fna_hopper/collective/fna_fusion.hpp:81
Functionget_window_start
csrc/include/natten/cuda/fna_blackwell/collective/fna_fusion.hpp:127
Methodget_workspace_size
Gets the workspace size
csrc/include/natten/cuda/fmha_blackwell/device/fmha_sm100.hpp:96
Methodget_workspace_size
Gets the workspace size
csrc/include/natten/cuda/fmha_blackwell/device/fmha_bwd_sm100.hpp:298
Methodget_workspace_size
csrc/include/natten/cuda/fmha_blackwell/kernel/sm100_fmha_fwd_kernel_tma_warpspecialized.hpp:241
Methodget_workspace_size
csrc/include/natten/cuda/fmha_blackwell/kernel/fmha_kernel_bwd_convert.hpp:80
← previousnext →1,301–1,400 of 2,059, ranked by callers