MCPcopy Create free account

hub / github.com/Tencent/hpc-ops / functions

Functions10,542 in github.com/Tencent/hpc-ops

↓ 29 callersFunctionpartition_shape_C
3rd/cutlass/include/cute/atom/mma_atom.hpp:561
↓ 28 callersFunctionbasis_get
3rd/cutlass/include/cute/numeric/arithmetic_tuple.hpp:262
↓ 28 callersFunctioncompute_epilogue
Returns whether the block assigned this work should compute the epilogue for the corresponding output tile. For the case of stream-K, this should only
3rd/cutlass/include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp:583
↓ 28 callersMethodcount
3rd/cutlass/include/cutlass/epilogue/threadblock/predicated_tile_iterator_params.h:84
↓ 28 callersMethodmake_fragment_A
3rd/cutlass/include/cute/atom/mma_atom.hpp:146
↓ 28 callersFunctionreverse
3rd/cutlass/include/cute/algorithm/tuple_algorithms.hpp:1042
↓ 28 callersFunctionsynclog_setup
3rd/cutlass/include/cutlass/arch/synclog.hpp:279
↓ 27 callersFunctionf
(v, w=11)
benchmark/fuse_allreduce_rmsorm/benchmark_fuse_allreduce_rmsnorm.py:393
↓ 27 callersMethodinitial_work_tile_info
3rd/cutlass/include/cutlass/gemm/kernel/sm90_tile_scheduler_group.hpp:614
↓ 27 callersFunctionpartition_shape_A
3rd/cutlass/include/cute/atom/mma_atom.hpp:588
↓ 27 callersMethodset_kgroup_index
Notify the iterator which k-group it is currently pointing to. This does not advance the iterator. Rather, it overrides its internal tracking with co
3rd/cutlass/include/cutlass/conv/warp/mma_depthwise_simt_tile_iterator.h:244
↓ 27 callersMethodtransform
Transform the mma operands to the required types
3rd/cutlass/include/cutlass/gemm/warp/mma_simt.h:251
↓ 26 callersMethodones
Constructs a matrix from a uniform element 1
3rd/cutlass/include/cutlass/matrix.h:125
↓ 26 callersMethodset_iteration_index
Overrides the internal iteration index
3rd/cutlass/include/cutlass/transform/threadblock/predicated_tile_access_iterator.h:238
↓ 26 callersMethodset_k_partition
Functionally required for serial reduction in the epilogue
3rd/cutlass/include/cutlass/epilogue/thread/conversion_op.h:97
↓ 26 callersFunctiontiled_product
3rd/cutlass/include/cute/layout.hpp:1696
↓ 25 callersFunctionmake_coord_tensor
3rd/cutlass/include/cute/tensor_impl.hpp:482
↓ 25 callersMethodmake_fragment_B
3rd/cutlass/include/cute/atom/mma_atom.hpp:172
↓ 25 callersFunctionmake_tmem_copy
3rd/cutlass/include/cute/atom/copy_traits_sm100.hpp:286
↓ 25 callersFunctionpartition_shape_B
3rd/cutlass/include/cute/atom/mma_atom.hpp:601
↓ 25 callersMethodrelease
Releases ownership of the managed object, if any
3rd/cutlass/include/cutlass/platform/platform.h:799
↓ 25 callersFunctiontransform_apply
3rd/cutlass/include/cute/algorithm/tuple_algorithms.hpp:125
↓ 24 callersFunctionapply
3rd/cutlass/include/cute/algorithm/tuple_algorithms.hpp:71
↓ 24 callersMethodgroups
3rd/cutlass/include/cutlass/gemm/group_array_problem_shape.hpp:97
↓ 24 callersMethoditers_per_tile
Number of iterations per output tile
3rd/cutlass/include/cutlass/gemm/threadblock/threadblock_swizzle_streamk.h:180
↓ 24 callersMethodpartition_C
3rd/cutlass/include/cute/atom/mma_atom.hpp:465
↓ 23 callersFunctionadd
3rd/cutlass/include/cute/numeric/real.hpp:40
↓ 23 callersFunctionblock_id_in_cluster
Returns the relative dim3 block rank local to the cluster.
3rd/cutlass/include/cute/arch/cluster_sm90.hpp:126
↓ 23 callersMethodget_thr_layout_vmnk
3rd/cutlass/include/cute/atom/mma_atom.hpp:233
↓ 23 callersFunctionprint
3rd/cutlass/include/cute/util/print.hpp:64
↓ 23 callersFunctionreduce
3rd/cutlass/include/cute/algorithm/tensor_reduce.hpp:45
↓ 23 callersMethodsum
Returns the sum of all elements
3rd/cutlass/include/cutlass/coord.h:309
↓ 22 callersMethodadvance
3rd/cutlass/include/cutlass/conv/threadblock/conv2d_tile_iterator.h:172
↓ 22 callersFunctionalignment_for_swizzle
3rd/cutlass/include/cutlass/detail/layout.hpp:418
↓ 22 callersFunctionconst_max
3rd/cutlass/include/cutlass/fast_math.h:757
↓ 22 callersFunctioncutlass_get_smem_pointer
CUTLASS helper to get SMEM pointer
3rd/cutlass/include/cutlass/arch/memory_sm75.h:63
↓ 22 callersFunctionmake_tiled_copy_D
3rd/cutlass/include/cute/atom/copy_atom.hpp:584
↓ 22 callersFunctionpipeline_check_is_consumer
3rd/cutlass/include/cutlass/pipeline/sm90_pipeline.hpp:80
↓ 22 callersMethodretile_S
3rd/cutlass/include/cute/atom/copy_atom.hpp:386
↓ 21 callersFunctioncluster_shape
Returns the dim3 cluster shape.
3rd/cutlass/include/cute/arch/cluster_sm90.hpp:140
↓ 21 callersFunctionflatten
3rd/cutlass/include/cute/layout.hpp:529
↓ 21 callersMethodmma
3rd/cutlass/include/cutlass/gemm/collective/sm100_mma_warpspecialized.hpp:648
↓ 21 callersFunctionunwrap
3rd/cutlass/include/cute/algorithm/tuple_algorithms.hpp:514
↓ 21 callersMethodvalid
Returns true if the current coordinate is within the filter tensor W
3rd/cutlass/include/cutlass/conv/threadblock/conv3d_fprop_filter_tile_access_iterator_optimized.h:227
↓ 20 callersFunctionconstexpr
Input A: float_e4m3_t ; Input B: float_e5m2_t
3rd/cutlass/include/cute/arch/mma_sm90.hpp:685
↓ 20 callersMethodget
Returns a pointer
3rd/cutlass/include/cutlass/transform/threadblock/regular_scale_bias_vector_access_iterator.h:129
↓ 20 callersMethodget
3rd/cutlass/include/cute/pointer_base.hpp:157
↓ 20 callersFunctionlaunch_dependent_grids
Issuing the launch_dependents instruction hints a dependent kernel to launch earlier launch_dependents doesn't impact the functionality but the perfor
3rd/cutlass/include/cutlass/arch/grid_dependency_control.h:84
↓ 20 callersFunctionload_with_byte_offset
Loads a fragment from memory with additional logical offset
3rd/cutlass/include/cutlass/gemm/warp/mma_tensor_op_tile_iterator.h:385
↓ 20 callersFunctionmake_tiled_copy_S
3rd/cutlass/include/cute/atom/copy_atom.hpp:573
↓ 20 callersFunctionmake_tiled_copy_impl
3rd/cutlass/include/cute/atom/copy_atom.hpp:408
↓ 20 callersMethodpartition_fragment_A
3rd/cutlass/include/cute/atom/mma_atom.hpp:506
↓ 20 callersFunctionreal
3rd/cutlass/include/cutlass/complex.h:393
↓ 20 callersMethodstore_tail
3rd/cutlass/include/cutlass/epilogue/collective/detail.hpp:514
↓ 20 callersFunctionstore_with_pointer_offset
Stores a fragment to memory at the location pointed to by the iterator
3rd/cutlass/include/cutlass/gemm/warp/mma_simt_tile_iterator.h:507
↓ 19 callersFunctionfast_divmod
* Find quotient and remainder using device-side intrinsics */
3rd/cutlass/include/cutlass/fast_math.h:274
↓ 19 callersFunctionfast_sqrt
3rd/cutlass/include/cutlass/fast_math.h:858
↓ 19 callersMethodget
Value
3rd/cutlass/include/cute/container/array_subbyte.hpp:161
↓ 19 callersFunctionmake_identity_layout
3rd/cutlass/include/cute/layout.hpp:483
↓ 19 callersFunctionmake_tiled_copy
3rd/cutlass/include/cute/atom/copy_atom.hpp:493
↓ 19 callersFunctionprepend
3rd/cutlass/include/cute/algorithm/tuple_algorithms.hpp:816
↓ 19 callersFunctiontma_desc_commit_group
3rd/cutlass/include/cute/arch/copy_sm90_tma.hpp:1235
↓ 19 callersFunctiontma_desc_wait_group
Wait until all TMA descriptor previously issued are safe to be modified after tma_desc_commit_group()
3rd/cutlass/include/cute/arch/copy_sm90_tma.hpp:1263
↓ 18 callersFunctionadd_pointer_offset
3rd/cutlass/include/cutlass/epilogue/threadblock/shared_load_iterator.h:67
↓ 18 callersMethoddata
Returns the pointer to referenced data
3rd/cutlass/include/cutlass/tensor_ref.h:268
↓ 18 callersFunctionget_gmem_layout
3rd/cutlass/include/cutlass/detail/collective/mixed_input_utils.hpp:527
↓ 18 callersFunctionget_logical_ptr
3rd/cutlass/include/cutlass/detail/collective/mixed_input_utils.hpp:506
↓ 18 callersMethodinverse
3rd/cutlass/include/cutlass/layout/matrix.h:114
↓ 18 callersMethodis_source_needed
3rd/cutlass/include/cutlass/epilogue/collective/default_epilogue.hpp:148
↓ 18 callersFunctionmake_fragment_like
3rd/cutlass/include/cute/layout.hpp:453
↓ 18 callersFunctionmake_tiled_copy_A
3rd/cutlass/include/cute/atom/copy_atom.hpp:422
↓ 18 callersFunctionmax_common_layout
3rd/cutlass/include/cute/layout.hpp:1382
↓ 18 callersMethodraw
Accesses raw internal state
3rd/cutlass/include/cutlass/half.h:463
↓ 18 callersFunctiontile_shape
3rd/cutlass/include/cute/atom/mma_atom.hpp:626
↓ 17 callersFunctionabs
3rd/cutlass/include/cutlass/complex.h:441
↓ 17 callersMethodbarrier
(self, channel: int = 0, timeout_ms: int = 0)
hpc/multicast_handle.py:198
↓ 17 callersFunctionblocked_product
3rd/cutlass/include/cute/layout.hpp:1732
↓ 17 callersFunctioncanonical_lane_idx
Returns a lane index in the warp. The threads in warp may not be convergent
3rd/cutlass/include/cutlass/cutlass.h:115
↓ 17 callersFunctionfence_view_async_tmem_load
3rd/cutlass/include/cutlass/arch/barrier.h:923
↓ 17 callersFunctionfill
3rd/cutlass/include/cute/algorithm/fill.hpp:45
↓ 17 callersMethodis_source_needed
Determine if the source is needed. May return false if
3rd/cutlass/include/cutlass/epilogue/threadblock/epilogue_with_broadcast.h:122
↓ 17 callersMethodload_init
3rd/cutlass/include/cutlass/gemm/collective/sm100_mma_warpspecialized.hpp:495
↓ 17 callersFunctionmax_common_vector
3rd/cutlass/include/cute/layout.hpp:1409
↓ 17 callersFunctionnorm
3rd/cutlass/include/cutlass/complex.h:479
↓ 17 callersFunctionstride
3rd/cutlass/include/cute/tensor_impl.hpp:541
↓ 17 callersFunctiontensor
3rd/cutlass/include/cute/tensor_impl.hpp:508
↓ 16 callersFunctionabs
3rd/cutlass/include/cute/numeric/math.hpp:65
↓ 16 callersMethodclear
Efficient clear method
3rd/cutlass/include/cutlass/array.h:360
↓ 16 callersFunctionconj
3rd/cutlass/include/cutlass/complex.h:566
↓ 16 callersFunctioncp_async_wait<0>
3rd/cutlass/include/cutlass/arch/memory_sm80.h:461
↓ 16 callersFunctionfold
3rd/cutlass/include/cute/algorithm/tuple_algorithms.hpp:387
↓ 16 callersFunctionget_work_k_tile_start
3rd/cutlass/include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp:698
↓ 16 callersFunctionimag
3rd/cutlass/include/cutlass/complex.h:402
↓ 16 callersMethodis_host_problem_shape_available
3rd/cutlass/include/cutlass/gemm/group_array_problem_shape.hpp:125
↓ 16 callersFunctionload_with_byte_offset
Loads a fragment from memory with additional logical offset
3rd/cutlass/include/cutlass/gemm/warp/mma_complex_tensor_op_tile_iterator_sm80.h:253
↓ 16 callersFunctionmake_utccp_copy
3rd/cutlass/include/cute/atom/copy_traits_sm100.hpp:3777
↓ 16 callersMethodstride
Returns the stride of the layout
3rd/cutlass/include/cutlass/layout/tensor_op_multiplicand_sm80.h:127
↓ 16 callersFunctionsynclog_emit_tma_load
3rd/cutlass/include/cutlass/arch/synclog.hpp:778
↓ 16 callersFunctionthr_size
3rd/cutlass/include/cute/atom/mma_atom.hpp:644
↓ 16 callersFunctiontuple_cat
3rd/cutlass/include/cute/container/tuple.hpp:291
← previousnext →201–300 of 10,542, ranked by callers