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

Methodto_gemm_args

src/all_gather/gemm_v3_ag_kernel.hpp:147

Methodto_gemm_args

src/all_gather/gemm_v2_ag_kernel.hpp:159

Methodto_gemm_args

src/reduce_scatter/gemm_v2_reduce_scatter.hpp:448

Methodto_gemm_args

src/reduce_scatter/gemm_v3_reduce_scatter.hpp:373

Methodto_gemm_args

include/flux/cuda/gemm_impls/gemm_grouped_impl.hpp:145

Methodto_gemm_args

include/flux/cuda/gemm_impls/gemm_v2_impl.hpp:358

Methodto_gemm_args_impl

src/comm_none/gemm_v3_comm_none.hpp:72

Methodto_gemm_args_impl

src/comm_none/gemm_v2_comm_none.hpp:113

Methodto_gemm_args_impl

src/all_gather/gemm_v3_ag_kernel.hpp:84

Methodto_gemm_args_impl

src/all_gather/gemm_v2_ag_kernel.hpp:101

Methodto_gemm_args_impl

src/reduce_scatter/gemm_v2_reduce_scatter.hpp:293

Methodto_gemm_args_impl

src/reduce_scatter/gemm_v3_reduce_scatter.hpp:239

Methodto_gemm_meta

include/flux/gemm_meta.h:367

Methodto_make_expr

include/flux/flux.h:1012

Functionto_std_tuple

include/flux/flux.h:630

Methodto_string_topk

src/ths_op/ths_op.cc:129

Methodto_tile_idx

src/reduce_scatter/reduce_scatter_kernel.hpp:447

include/flux/cuda/cutlass_v3_builder.hpp:90

Methodto_underlying_arguments

src/all_gather/sm90_all_gather_gemm_tile_scheduler.hpp:80

Methodto_underlying_arguments

Convert to underlying arguments. In this case, a simple copy for the aliased type.

src/all_gather/sm90_all_gather_gemm_tma_warpspecialized_cooperative.hpp:210

Methodto_underlying_arguments

src/reduce_scatter/epilogue_reduce_scatter.hpp:127

Methodto_underlying_arguments

src/reduce_scatter/epilogue_evt.hpp:129

Methodto_underlying_arguments

src/reduce_scatter/epilogue_vectorized_reduce_scatter.hpp:137

Methodto_underlying_arguments

src/reduce_scatter/epilogue_evt_nvshmem.hpp:105

Methodto_underlying_arguments

src/reduce_scatter/epilogue_nvshmem_reduce_scatter.hpp:149

Methodto_underlying_arguments

src/reduce_scatter/sm90_epilogue_evt.hpp:134

Methodto_underlying_arguments

Convert to underlying arguments. In this case, a simple copy for the aliased type.

src/reduce_scatter/sm90_gemm_tma_warpspecialized_cooperative_reduce_scatter.hpp:170

Methodto_underlying_arguments

src/reduce_scatter/sm90_reduce_scatter_utils.hpp:142

Methodto_underlying_arguments

src/reduce_scatter/gemmk_visitor_load.hpp:75

Methodto_underlying_arguments

src/reduce_scatter/tile_scheduler/tile_mappings.hpp:48

Methodto_underlying_arguments

src/reduce_scatter/tile_scheduler/tile_mappings.hpp:89

Methodto_underlying_arguments

src/reduce_scatter/tile_scheduler/sm90_tile_scheduler_reduce_scatter.hpp:53

Methodto_underlying_arguments

src/reduce_scatter/tile_scheduler/sm90_tile_scheduler_reduce_scatter.hpp:124

Functiontopo_numa_local_world_size

src/ths_op/topo_utils.cc:354

Functiontopo_nvlink_local_world_size

src/ths_op/topo_utils.cc:333

Functiontorch_allclose

(x, y, rtol, atol)

python/flux/util.py:86

Functiontorch_dtype_size

src/ths_op/ths_op.cc:71

Methodtransposed_problem

Returns arguments for the transposed problem

src/all_gather/sm80_all_gather_gemm.hpp:263

Functiontuple_cartesian_product_impl

include/flux/flux.h:303

Functiontuple_cartesian_product_two

include/flux/flux.h:278

Functiontuple_for_each

include/flux/flux.h:241

Functiontuple_has_elem_i

include/flux/flux.h:428

Functiontuple_length

include/flux/flux.h:185

Functiontuple_replace_item

include/flux/flux.h:202

Functiontuple_return_if

include/flux/flux.h:359

Functiontuple_return_if_impl

include/flux/flux.h:335

Functiontuple_split

include/flux/flux.h:171

Functiontuple_split_slice

include/flux/flux.h:388

Functionuniform_initialize

src/ths_op/helper_ops.cc:32

Methodunify_type

include/flux/gemm_meta.h:160

Methodunify_type

include/flux/gemm_meta.h:247

Lightweight update given a subset of arguments.

src/all_gather/sm80_all_gather_gemm.hpp:495

Lightweight update given a subset of arguments.

src/reduce_scatter/gemm_v2_reduce_scatter.hpp:72

src/reduce_scatter/reduce_scatter_kernel.hpp:1999

Lightweight update given a subset of arguments.

src/reduce_scatter/gemmk_universal_with_visitor_streamk.h:446

Methoduse_1d_ring_or_not

src/reduce_scatter/ths_op/gemm_reduce_scatter.cc:165

Methoduse_p2p_read_or_not

src/reduce_scatter/ths_op/gemm_reduce_scatter.cc:176

src/reduce_scatter/epilogue_evt.hpp:240

src/reduce_scatter/epilogue_evt_nvshmem.hpp:230

src/reduce_scatter/sm90_epilogue_evt.hpp:215

src/reduce_scatter/gemmk_visitor_load.hpp:142

Functionwait_async

src/reduce_scatter/reduce_scatter_kernel.hpp:213

Methodwait_cross_numa_done

src/all_gather/ths_op/all_gather_gemm_kernel_crossnode.cc:644

src/reduce_scatter/reduce_scatter_kernel.hpp:124

include/flux/cuda/system_barrier.hpp:79

Methodwait_eq_reset

include/flux/cuda/system_barrier.hpp:92

Methodwait_eq_reset

include/flux/cuda/system_barrier.hpp:141

include/flux/cuda/system_barrier.hpp:64

Methodwait_ready

src/all_gather/ths_op/transfers.hpp:43

Methodwait_ready

src/all_gather/ths_op/all_gather_gemm_kernel.cc:807

Functionwith_torch_deterministic

(mode: bool, warn_only: bool = True)

python/flux/util.py:130

Methodwork_queue_flags

src/reduce_scatter/reduce_scatter_kernel.hpp:315

Functionwrapper

(*kargs, **kwargs)

Methodzero_buffers

src/reduce_scatter/ths_op/gemm_reduce_scatter.cc:648

Method~AGKernel

src/all_gather/ths_op/all_gather_gemm_kernel.cc:252

Method~AGKernelCrossNode

src/all_gather/ths_op/all_gather_gemm_kernel_crossnode.cc:292

Method~AppendCloseIpcMemHandleCallbackRAII

src/ths_op/ths_op.cc:99

Method~CheckFail

Destructor that throws an exception with the accumulated error message.

include/flux/flux.h:103

Method~DynamicLibrary

src/cuda/nvml_stub.cc:98

Method~DynamicLibrary

src/cuda/cuda_stub.cc:94

Method~GemmOperatorBase

include/flux/gemm_operator_base.h:88

Method~GemmOperatorBaseDefaultImplMixin

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

src/reduce_scatter/ths_op/gemm_reduce_scatter.cc:279

Method~GpuTimer

include/flux/cuda/cuda_common.h:176

← previous701–784 of 784, ranked by callers