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

Methodis_C_load_needed

src/reduce_scatter/sm90_epilogue_evt.hpp:178

Functionlazy_init_buffer_tensor

src/ths_op/ths_op.cc:405

Methodld_acquire

Load flag, as a strong acquire operation (int specialization)

include/flux/cuda/system_barrier.hpp:36

include/flux/flux.h:894

Functionload_tune_config_from_file

src/cuda/op_registry_proto_utils.cc:165

Functionload_tuning_record

src/ths_op/ths_op.cc:112

Functionlocal_rank_to_global_rank

include/flux/utils.h:36

Methodlocal_red

include/flux/cuda/memory_utils.hpp:193

Methodlocal_red

include/flux/cuda/memory_utils.hpp:223

Methodlower_name

include/flux/flux.h:945

src/comm_none/test/test_gemm_only.cc:86

src/reduce_scatter/test/test_gemm_rs.cc:142

Methodmake_gather_rs_meta

include/flux/gemm_meta.h:259

Methodmake_gemm_meta

include/flux/gemm_meta.h:354

Functionmake_index_seq_tuple

include/flux/flux.h:328

Methodmake_space_gemm_meta

include/flux/gemm_meta.h:424

Functionmemset_continous_kernel

src/reduce_scatter/reduce_scatter_kernel.hpp:239

Methodnum_fields

include/flux/flux.h:950

Functionnvshmemi_bootstrap_plugin_init

pynvshmem/bootstrap/torch/bootstrap_torch.cpp:122

Functionoffset_index_sequence

include/flux/flux.h:157

Methodoperator()

src/all_gather/sm80_all_gather_gemm.hpp:1028

Methodoperator()

src/all_gather/sm90_all_gather_gemm_tma_warpspecialized_cooperative.hpp:387

Methodoperator()

src/reduce_scatter/epilogue_reduce_scatter.hpp:159

Methodoperator()

Runs the kernel using initialized state.

src/reduce_scatter/gemm_v2_reduce_scatter.hpp:116

Methodoperator()

src/reduce_scatter/reduce_scatter_kernel.hpp:164

Methodoperator()

src/reduce_scatter/reduce_scatter_kernel.hpp:2004

Methodoperator()

src/reduce_scatter/epilogue_vectorized_reduce_scatter.hpp:170

Methodoperator()

src/reduce_scatter/epilogue_nvshmem_reduce_scatter.hpp:182

Methodoperator()

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:963

Methodoperator()

src/reduce_scatter/sm90_gemm_tma_warpspecialized_cooperative_reduce_scatter.hpp:335

Functionoperator<

include/flux/flux.h:918

Methodoperator=

src/cuda/nvml_stub.cc:77

Methodoperator=

src/cuda/cuda_stub.cc:69

Functionoperator==

include/flux/flux.h:925

src/cuda/utils.cc:27

pad `sz` to the minimum multiple of `pad`

include/flux/flux.h:623

Methodpartition

src/reduce_scatter/visitor_2x_bsr.hpp:140

Functionpathlib_wrapper

Functionprint_cute_per_kernel

include/flux/cuda/cuda_common_device.hpp:81

Functionprint_per_block_

include/flux/cuda/cuda_common_device.hpp:61

Methodprocess_tile

src/all_gather/sm80_all_gather_gemm.hpp:787

Methodprocess_tile

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:768

Methodprofiling

src/comm_none/ths_op/gemm_only.cc:274

Functionptr_offset

calculate void* ptr with offset of bytes

include/flux/flux.h:612

Functionput_tensor_on_stream

( dst_tensor: torch.Tensor, src_tensor: torch.Tensor, peer: int, stream: CUDA_STREAM_TYPE, )

python/flux/pynvshmem/_wrapper.py:20

Functionputmem_on_stream

pynvshmem/src/functions.cpp:80

Methodpybind_name

pynvshmem/tools/autogen_pybind.py:148

Methodpybind_name_wrapper

pynvshmem/tools/autogen_pybind.py:154

Functionpyflux_barrier_all_on_stream

src/ths_op/flux_shm.cc:216

pynvshmem/src/functions.cpp:61

Functionrank_shift

include/flux/utils.h:82

Methodrecast_backto_bf16

src/reduce_scatter/epilogue_evt_nvshmem.hpp:247

Methodred_release

Reduce into flag, with release pattern (int specialization)

include/flux/cuda/system_barrier.hpp:46

Methodreduce_buffer_tile

src/reduce_scatter/reduce_scatter_kernel.hpp:368

Methodregister_creator

include/flux/op_registry.h:140

Methodregister_dispatcher

include/flux/op_registry.h:173

FunctionreleaseCpuDataCallback

src/ths_op/ths_op.cc:89

Methodreset_signals

src/all_gather/ths_op/all_gather_gemm_kernel_crossnode.cc:381

(string, pattern)

pynvshmem/tools/autogen_pybind.py:111

Methodring_all_gather

src/all_gather/ths_op/all_gather_gemm_kernel.cc:889

Methodring_all_gather

src/all_gather/ths_op/all_gather_gemm_kernel_crossnode.cc:808

Methodring_all_gather_reorder

src/all_gather/ths_op/all_gather_gemm_kernel.cc:895

Methodring_all_gather_reorder

src/all_gather/ths_op/all_gather_gemm_kernel_crossnode.cc:814

Methodring_all_gather_run

src/all_gather/ths_op/all_gather_gemm_kernel.cc:907

Methodring_all_gather_run

src/all_gather/ths_op/all_gather_gemm_kernel_crossnode.cc:826

Runs the kernel using initialized state.

src/reduce_scatter/gemm_v2_reduce_scatter.hpp:82

src/reduce_scatter/reduce_scatter_kernel.hpp:514

src/reduce_scatter/reduce_scatter_kernel.hpp:643

src/reduce_scatter/reduce_scatter_kernel.hpp:818

src/reduce_scatter/reduce_scatter_kernel.hpp:942

src/reduce_scatter/reduce_scatter_kernel.hpp:1084

src/reduce_scatter/reduce_scatter_kernel.hpp:1233

src/reduce_scatter/reduce_scatter_kernel.hpp:1323

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

Functionrun_gemm_only

src/comm_none/test/test_gemm_only.cc:32

Functionrun_per_segment_kernel

src/reduce_scatter/reduce_scatter_kernel.hpp:1443

Functionrun_per_segment_kernel_multinode

src/reduce_scatter/reduce_scatter_kernel.hpp:1646

Functionrun_per_segment_kernel_tp8

src/reduce_scatter/reduce_scatter_kernel.hpp:1508

Functionrun_per_segment_with_cudaMemcpyAsync

src/reduce_scatter/reduce_scatter_kernel.hpp:1863

Methodrun_per_segment_with_cuda_core

src/reduce_scatter/reduce_scatter_kernel.hpp:2009

Functionrun_perf

(expname, warmups, iters, func, sync_per_iter=False)

test/utils.py:106

Methodseparate_reduction

src/all_gather/sm80_all_gather_gemm.hpp:749

Methodseparate_reduction

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:730

Methodset_ready

src/all_gather/ths_op/transfers.hpp:28

Methodset_ready

src/all_gather/ths_op/all_gather_gemm_kernel.cc:798

Methodshare_accumulators

Share accumulators with peers

src/all_gather/sm80_all_gather_gemm.hpp:688

Methodshare_accumulators

Share accumulators with peers

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:661

Functionsleep_async

src/reduce_scatter/reduce_scatter_kernel.hpp:208

Methodsplit_slice_meta

include/flux/gemm_operator_base.h:63

src/reduce_scatter/epilogue_nvshmem_reduce_scatter.hpp:203

include/flux/cuda/system_barrier.hpp:27

Methodtid2coord

src/reduce_scatter/visitor_2x_bsr.hpp:133

src/ths_op/ths_op.cc:281

Methodto_argument_type

src/reduce_scatter/gemm_v2_reduce_scatter.hpp:484

Functionto_cuda_dtype

include/flux/cuda/cuda_common.h:75

Methodto_fp8_gemm_args_impl

src/comm_none/gemm_v2_comm_none.hpp:54

Methodto_fp8_gemm_args_impl

src/reduce_scatter/gemm_v2_reduce_scatter.hpp:389

Methodto_gather_rs_meta

include/flux/gemm_meta.h:265

Methodto_gemm_args

src/comm_none/gemm_v3_comm_none.hpp:126

Methodto_gemm_args

src/comm_none/gemm_v2_comm_none.hpp:163

← previous next →601–700 of 784, ranked by callers