Functions — github.com/bytedance/flux

CodeHub MCPcopy Create free account

hub / github.com/bytedance/flux / functions

Functions784 in github.com/bytedance/flux

⨍ Functions 784◇ Types & classes 309

Methodgemm_kernel

src/reduce_scatter/gemm_v2_reduce_scatter.hpp:268

Methodgemm_only

src/all_gather/ths_op/all_gather_gemm_kernel_crossnode.cc:637

Methodgemm_only_impl

src/all_gather/ths_op/all_gather_gemm_kernel_crossnode.cc:586

include/flux/op_registry.h:48

include/flux/op_registry.h:60

Methodget_all_records

src/ths_op/ths_op.cc:157

Methodget_args_workspace_size

src/reduce_scatter/gemm_v2_reduce_scatter.hpp:472

Methodget_barrier_workspace_size

Get the workspace size needed for barrier

src/all_gather/sm80_all_gather_gemm.hpp:338

Methodget_barrier_workspace_size

src/reduce_scatter/gemm_v2_reduce_scatter.hpp:458

Methodget_barrier_workspace_size

Get the workspace size needed for barrier template <class ThreadblockShape>

src/reduce_scatter/epilogue_evt_nvshmem.hpp:153

Methodget_barrier_workspace_size

Get the workspace size needed for barrier

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:285

Methodget_barrier_workspace_size

src/reduce_scatter/gemm_v3_reduce_scatter.hpp:378

Methodget_barrier_workspace_size

include/flux/gemm_operator_base.h:109

Methodget_barrier_workspace_size

include/flux/cuda/gemm_impls/gemm_operator_base_default_impl.hpp:92

Methodget_block_shape

src/all_gather/sm90_all_gather_gemm_tma_warpspecialized_cooperative.hpp:382

Methodget_block_shape

src/reduce_scatter/sm90_gemm_tma_warpspecialized_cooperative_reduce_scatter.hpp:330

Methodget_callbacks

src/reduce_scatter/epilogue_evt.hpp:386

Methodget_callbacks

src/reduce_scatter/epilogue_evt_nvshmem.hpp:342

Methodget_callbacks

src/reduce_scatter/gemmk_visitor_load.hpp:156

Functionget_consumer_store_callbacks

src/reduce_scatter/sm90_epilogue_evt.hpp:252

Functionget_cu_error_string

include/flux/cuda/cuda_stub.h:50

Methodget_current_work

src/reduce_scatter/tile_scheduler/sm90_tile_scheduler_reduce_scatter.hpp:159

Functionget_dist_env

(deterministic: bool = True)

python/flux/dist_utils.py:71

Methodget_func_attr

src/reduce_scatter/reduce_scatter_kernel.hpp:2175

Methodget_gemm_meta

src/all_gather/ths_op/all_gather_gemm_kernel_crossnode.cc:156

Methodget_grid_blocks

Returns the total number of thread blocks to launch

src/all_gather/sm80_all_gather_gemm.hpp:482

Methodget_grid_blocks

Returns the total number of thread blocks to launch

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:433

Methodget_grid_dims

Returns the grid extents in thread blocks to launch

src/all_gather/sm80_all_gather_gemm.hpp:489

Methodget_grid_dims

Returns the grid extents in thread blocks to launch

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:440

Methodget_grid_shape

Computes the kernel launch grid shape based on runtime parameters

src/all_gather/sm90_all_gather_gemm_tma_warpspecialized_cooperative.hpp:367

Methodget_grid_shape

Computes the kernel launch grid shape based on runtime parameters

src/reduce_scatter/sm90_gemm_tma_warpspecialized_cooperative_reduce_scatter.hpp:315

Functionget_inter_node_rs_group

python/flux/gemm_rs_sm80.py:52

Methodget_latest_record

src/ths_op/ths_op.cc:182

Functionget_local_rank_from_env

src/cuda/utils.cc:98

Functionget_local_world_size_from_env

src/cuda/utils.cc:88

Functionget_nvshmem_nodes

pynvshmem/tools/autogen_pybind.py:260

Methodget_optional_scale_tensor

src/comm_none/ths_op/gemm_only.cc:236

Methodget_optional_scale_tensor

src/all_gather/ths_op/all_gather_gemm_kernel.cc:236

Methodget_partials_workspace_size

Get the workspace size needed for intermediate partial sums

src/all_gather/sm80_all_gather_gemm.hpp:349

Methodget_partials_workspace_size

Get the workspace size needed for intermediate partial sums

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:296

Methodget_producer_load_callbacks

src/reduce_scatter/sm90_epilogue_evt.hpp:184

Functionget_rank_from_env

src/cuda/utils.cc:93

Methodget_rt_config

src/all_gather/ths_op/all_gather_gemm_kernel.cc:281

Methodget_rt_config

src/all_gather/ths_op/all_gather_gemm_kernel_crossnode.cc:837

Methodget_runtime_gemm_hparams

include/flux/cuda/gemm_impls/gemm_operator_base_default_impl.hpp:97

Functionget_sm_count

src/cuda/cuda_common.cc:60

Methodget_tile_offset

Obtains the calling threadblock's tiled coordinates for the given tile index

src/reduce_scatter/tile_scheduler/threadblock_swizzle_acrossnode.hpp:78

Methodget_tiled_shape

Returns the GEMM volume in thread block tiles

src/all_gather/sm80_all_gather_gemm.hpp:476

Methodget_tiled_shape

Returns the GEMM volume in thread block tiles

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:427

Methodget_tma_descriptor

Return TmaDescriptor/TensorMap

include/flux/cuda/memory_utils.hpp:388

Functionget_torch_prof_ctx

(do_prof: bool)

python/flux/util.py:113

Methodget_workspace_size

src/all_gather/sm90_all_gather_gemm_tma_warpspecialized_cooperative.hpp:307

Methodget_workspace_size

Returns the workspace size (in bytes) needed for these parameters

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:375

Methodget_workspace_size

src/reduce_scatter/sm90_epilogue_evt.hpp:148

Methodget_workspace_size

src/reduce_scatter/sm90_gemm_tma_warpspecialized_cooperative_reduce_scatter.hpp:255

Methodget_workspace_size

total workspace: args_workspace + gemm workspace

include/flux/gemm_operator_base.h:104

Methodget_workspace_size

include/flux/cuda/gemm_impls/gemm_operator_base_default_impl.hpp:78

Methodget_world

python/flux/dist_utils.py:61

Functionget_world_size_from_env

src/cuda/utils.cc:83

Methodglobal_rank_to_local_rank

include/flux/utils.h:54

Methodglobal_red

include/flux/cuda/memory_utils.hpp:64

Methodglobal_red

include/flux/cuda/memory_utils.hpp:80

Methodglobal_red

include/flux/cuda/memory_utils.hpp:99

Methodglobal_red

include/flux/cuda/memory_utils.hpp:116

Functionhas_heterogeneous_nvlink

src/ths_op/topo_utils.cc:326

Functionhas_heterogeneous_pcie

src/ths_op/topo_utils.cc:347

Methodhas_nvlink

src/reduce_scatter/ths_op/gemm_reduce_scatter.cc:160

Functionhas_nvswitch

src/ths_op/topo_utils.cc:319

src/reduce_scatter/epilogue_vectorized_reduce_scatter.hpp:281

src/reduce_scatter/epilogue_nvshmem_reduce_scatter.hpp:306

Functionindex_seq_split_slice

include/flux/flux.h:371

Functioninit_ag_kernel_ops

src/all_gather/ths_op/all_gather_gemm_kernel.cc:919

Methodinit_dp_tile_work

src/all_gather/sm80_all_gather_gemm.hpp:627

Methodinit_dp_tile_work

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:600

Functioninit_flux_shm

src/ths_op/flux_shm.cc:158

Methodinit_iterator_A

src/all_gather/sm80_all_gather_gemm.hpp:579

Methodinit_iterator_A

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:528

Methodinit_iterator_B

src/all_gather/sm80_all_gather_gemm.hpp:604

Methodinit_iterator_B

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:577

Functioninit_local_groups

test/utils.py:65

Methodinit_sk_tile_work

src/all_gather/sm80_all_gather_gemm.hpp:651

Methodinit_sk_tile_work

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:624

Functioninit_with_c10d_pg

pynvshmem/src/functions.cpp:26

Methodinit_workspace

Assign and initialize the specified workspace buffer. Assumes the memory allocated to workspace is at least as large as get_workspace_size().

src/all_gather/sm80_all_gather_gemm.hpp:431

Methodinit_workspace

Assign and initialize the specified workspace buffer. Assumes the memory allocated to workspace is at least as large as get_workspace_size().

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:382

Methodinitialize

Initializes GEMM state from arguments and workspace memory

src/reduce_scatter/gemm_v2_reduce_scatter.hpp:60

Methodinitialize_args_workspace

src/reduce_scatter/gemm_v2_reduce_scatter.hpp:477

Functioninitialize_inter_node_rs_group

(tp_group: dist.ProcessGroup, nnodes: int)

python/flux/gemm_rs_sm80.py:60

Methodinitialize_workspace

src/all_gather/sm90_all_gather_gemm_tma_warpspecialized_cooperative.hpp:323

Methodinitialize_workspace

src/reduce_scatter/sm90_epilogue_evt.hpp:154

Methodinitialize_workspace

src/reduce_scatter/sm90_gemm_tma_warpspecialized_cooperative_reduce_scatter.hpp:271

Factory invocation

src/all_gather/sm80_all_gather_gemm.hpp:1008

src/reduce_scatter/reduce_scatter_kernel.hpp:626

src/reduce_scatter/reduce_scatter_kernel.hpp:802

src/reduce_scatter/reduce_scatter_kernel.hpp:923

src/reduce_scatter/reduce_scatter_kernel.hpp:1066

src/reduce_scatter/reduce_scatter_kernel.hpp:1217

src/reduce_scatter/reduce_scatter_kernel.hpp:1305

src/reduce_scatter/reduce_scatter_kernel.hpp:1435

Factory invocation

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:943

← previous next →501–600 of 784, ranked by callers