MCPcopy Create free account

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

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

↓ 1,109 callersFunctionmake_tensor
3rd/cutlass/include/cute/tensor_impl.hpp:396
↓ 776 callersMethodat
Accesses an element by coordinate
3rd/cutlass/include/cutlass/matrix.h:148
↓ 740 callersFunctionmake_layout
3rd/cutlass/include/cute/layout.hpp:333
↓ 735 callersFunctionsize
3rd/cutlass/include/cute/layout.hpp:605
↓ 534 callersFunctioncopy
3rd/cutlass/include/cute/algorithm/copy.hpp:171
↓ 520 callersMethoddata
3rd/cutlass/include/cutlass/gemm/kernel/sm90_tile_scheduler.hpp:64
↓ 509 callersMethodcolumn
3rd/cutlass/include/cutlass/matrix.h:2094
↓ 502 callersFunctionrank
3rd/cutlass/include/cute/layout.hpp:614
↓ 475 callersMethodrow
3rd/cutlass/include/cutlass/matrix.h:229
↓ 427 callersFunctionmake_smem_ptr
3rd/cutlass/include/cute/pointer.hpp:165
↓ 399 callersFunctionshape
3rd/cutlass/include/cute/int_tuple.hpp:102
↓ 365 callersMethodsize
3rd/cutlass/include/cute/tensor_impl.hpp:202
↓ 333 callersMethodstride
Returns the stride of the layout
3rd/cutlass/include/cutlass/layout/matrix.h:119
↓ 326 callersMethodbegin
Before entry of the subtile load loop
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_tma_warpspecialized.hpp:132
↓ 320 callersMethodindex
3rd/cutlass/include/cutlass/pipeline/sm90_pipeline.hpp:188
↓ 320 callersFunctionsynclog_emit_wgmma_reg_smem
3rd/cutlass/include/cutlass/arch/synclog.hpp:898
↓ 320 callersFunctionsynclog_emit_wgmma_smem_smem
3rd/cutlass/include/cutlass/arch/synclog.hpp:916
↓ 264 callersMethodpartition_S
3rd/cutlass/include/cute/atom/copy_atom.hpp:366
↓ 257 callersFunctionlocal_tile
3rd/cutlass/include/cute/tensor_impl.hpp:1035
↓ 240 callersMethodget_slice
3rd/cutlass/include/cute/atom/mma_atom.hpp:357
↓ 236 callersFunctionelect_one_sync
Elect one thread in the warp. The elected thread gets its predicate set to true, all others obtain false.
3rd/cutlass/include/cute/arch/cluster_sm90.hpp:180
↓ 232 callersMethodwith
3rd/cutlass/include/cute/atom/mma_atom.hpp:76
↓ 229 callersMethodimag
Set the imaginary part of the complex number
3rd/cutlass/include/cutlass/complex.h:315
↓ 215 callersMethodreal
Set the real part of the complex number
3rd/cutlass/include/cutlass/complex.h:311
↓ 214 callersFunctionprint
3rd/cutlass/include/cute/atom/mma_atom.hpp:656
↓ 207 callersMethodpartition_D
3rd/cutlass/include/cute/atom/copy_atom.hpp:376
↓ 200 callersFunctionprint
3rd/cutlass/include/cutlass/conv/convnd_problem_shape.hpp:579
↓ 193 callersFunctionfilter_zeros
3rd/cutlass/include/cute/layout.hpp:915
↓ 188 callersFunctionappend
3rd/cutlass/include/cute/layout_composed.hpp:308
↓ 188 callersFunctiongemm
3rd/cutlass/include/cute/algorithm/gemm.hpp:68
↓ 185 callersFunctionto_underlying_arguments
3rd/cutlass/include/cutlass/gemm/kernel/sm100_tile_scheduler_stream_k.hpp:114
↓ 166 callersFunctionround_nearest
3rd/cutlass/include/cutlass/fast_math.h:151
↓ 164 callersMethodadd_tile_offset
3rd/cutlass/include/cutlass/epilogue/threadblock/shared_load_iterator_mixed.h:185
↓ 164 callersMethodget_tma_descriptor
Return TmaDescriptor/TensorMap
3rd/cutlass/include/cute/atom/copy_traits_sm90_tma.hpp:197
↓ 164 callersMethodget_tma_tensor
3rd/cutlass/include/cute/atom/copy_traits_sm90_tma.hpp:149
↓ 164 callersFunctionget_workspace_size
3rd/cutlass/include/cutlass/gemm/kernel/sm90_tile_scheduler.hpp:139
↓ 163 callersMethodshape
3rd/cutlass/include/cute/tensor_impl.hpp:198
↓ 159 callersMethoddata
3rd/cutlass/include/cutlass/epilogue/collective/detail.hpp:171
↓ 157 callersFunctionmin
3rd/cutlass/include/cute/int_tuple.hpp:154
↓ 152 callersFunctiontransform
3rd/cutlass/include/cute/algorithm/tuple_algorithms.hpp:205
↓ 149 callersFunctioncosize
3rd/cutlass/include/cute/layout.hpp:662
↓ 148 callersFunctionadd_tile_offset
3rd/cutlass/include/cutlass/epilogue/threadblock/shared_load_iterator.h:155
↓ 147 callersFunctionElement
Unpacks an element from memory
3rd/cutlass/include/cutlass/subbyte_reference.h:187
↓ 147 callersMethodconsumer_release
3rd/cutlass/include/cutlass/pipeline/sm90_pipeline.hpp:1159
↓ 144 callersFunctioncast_smem_ptr_to_uint
CUTE helper to cast SMEM pointer to unsigned
3rd/cutlass/include/cute/arch/util.hpp:91
↓ 143 callersFunctioncan_implement
3rd/cutlass/include/cutlass/gemm/kernel/sm100_tile_scheduler_stream_k.hpp:220
↓ 143 callersFunctiontiled_divide
3rd/cutlass/include/cute/layout.hpp:1619
↓ 141 callersMethodconsumer_wait
3rd/cutlass/include/cutlass/pipeline/sm90_pipeline.hpp:1154
↓ 139 callersFunctionbitcast
3rd/cutlass/include/cutlass/exmy_base.h:997
↓ 139 callersMethoddata
3rd/cutlass/include/cute/container/array.hpp:93
↓ 138 callersMethodproducer_acquire
3rd/cutlass/include/cutlass/pipeline/sm90_pipeline.hpp:1109
↓ 137 callersMethodconsumer_try_wait
3rd/cutlass/include/cutlass/pipeline/sm90_pipeline.hpp:1145
↓ 137 callersMethodis_valid
3rd/cutlass/include/cutlass/gemm/kernel/static_tile_scheduler.hpp:61
↓ 136 callersMethodstride
3rd/cutlass/include/cute/tensor_impl.hpp:210
↓ 135 callersFunctioncoalesce
3rd/cutlass/include/cute/layout.hpp:865
↓ 133 callersFunctionidx2crd
3rd/cutlass/include/cute/stride.hpp:183
↓ 132 callersFunctiontile_to_shape
3rd/cutlass/include/cute/layout.hpp:1776
↓ 129 callersMethodclear
< Efficiently disables all accesses guarded by mask
3rd/cutlass/include/cutlass/epilogue/threadblock/predicated_tile_iterator_conv.h:193
↓ 128 callersFunctionceil_div
Returns the ceiling of (a / b)
3rd/cutlass/include/cutlass/fast_math.h:219
↓ 128 callersMethodlayout
Returns the layout object
3rd/cutlass/include/cutlass/thread/matrix.h:125
↓ 122 callersFunctionrepeat_like
3rd/cutlass/include/cute/algorithm/tuple_algorithms.hpp:734
↓ 121 callersMethodget
Returns a pointer to the managed object or nullptr if no object is owned.
3rd/cutlass/include/cutlass/platform/platform.h:796
↓ 117 callersFunctionmake_gmem_ptr
3rd/cutlass/include/cute/pointer.hpp:102
↓ 114 callersFunctionbits_to_bytes
3rd/cutlass/include/cutlass/numeric_size.h:70
↓ 114 callersMethodfetch_next_work
3rd/cutlass/include/cutlass/gemm/kernel/sm100_tile_scheduler.hpp:454
↓ 114 callersMethodload
3rd/cutlass/include/cutlass/gemm/collective/sm100_mma_warpspecialized.hpp:585
↓ 114 callersFunctionprefetch_tma_descriptor
3rd/cutlass/include/cute/arch/copy_sm90_desc.hpp:302
↓ 114 callersFunctionraw_pointer_cast
3rd/cutlass/include/cute/container/array_subbyte.hpp:333
↓ 111 callersFunctionmax
3rd/cutlass/include/cute/int_tuple.hpp:134
↓ 110 callersMethodw
Accesses the real part of the quaternion
3rd/cutlass/include/cutlass/quaternion.h:188
↓ 109 callersMethodmn
Obtains a Coord<2> from GemmCoord
3rd/cutlass/include/cutlass/gemm_coord.h:162
↓ 108 callersMethodset_iteration_index
Overrides the internal iteration index
3rd/cutlass/include/cutlass/conv/threadblock/conv2d_tile_iterator.h:108
↓ 105 callersFunctionmake_tma_copy
3rd/cutlass/include/cute/atom/copy_traits_sm90_tma.hpp:1315
↓ 103 callersMethodproducer_tail
Prevents early exit of producer blocks in Cluster. This should be called once before kernel exits.
3rd/cutlass/include/cutlass/pipeline/sm90_pipeline.hpp:1128
↓ 100 callersFunctionprint
3rd/cutlass/include/cute/layout.hpp:1918
↓ 99 callersMethodfloat
Converts to float
3rd/cutlass/include/cutlass/half.h:421
↓ 99 callersFunctionrun
3rd/cutlass/include/cutlass/transform/kernel/sm90_sparse_gemm_compressor.hpp:218
↓ 97 callersFunctionsize
3rd/cutlass/include/cute/atom/mma_atom.hpp:635
↓ 93 callersMethodarrive
Barrier arrive on local smem
3rd/cutlass/include/cutlass/arch/barrier.h:376
↓ 93 callersMethodclear_mask
< Efficiently disables all accesses guarded by mask
3rd/cutlass/include/cutlass/epilogue/threadblock/predicated_tile_iterator.h:1050
↓ 91 callersFunctionlayout
3rd/cutlass/include/cute/tensor_impl.hpp:523
↓ 89 callersMethodlayout
3rd/cutlass/include/cutlass/epilogue/threadblock/output_iterator_parameter.h:74
↓ 89 callersFunctionwarpgroup_fence_operand
3rd/cutlass/include/cute/arch/mma_sm90_gmma.hpp:86
↓ 87 callersFunctionBase
3rd/cutlass/include/cutlass/epilogue/threadblock/predicated_tile_iterator_conv.h:171
↓ 84 callersFunctionzipped_divide
3rd/cutlass/include/cute/layout.hpp:1608
↓ 83 callersMethodproducer_try_acquire
3rd/cutlass/include/cutlass/pipeline/sm90_pipeline.hpp:1105
↓ 82 callersFunctionelem_less
3rd/cutlass/include/cute/int_tuple.hpp:883
↓ 82 callersMethodvalue
3rd/cutlass/include/cute/numeric/arithmetic_tuple.hpp:228
↓ 81 callersFunctionselect_cluster_shape
3rd/cutlass/include/cutlass/detail/cluster.hpp:49
↓ 80 callersFunctioncomposition
3rd/cutlass/include/cute/layout.hpp:1134
↓ 80 callersMethodget_thread_slice
3rd/cutlass/include/cute/atom/mma_atom.hpp:367
↓ 80 callersFunctionmma
3rd/cutlass/include/cutlass/gemm/collective/sm100_mma_array_warpspecialized.hpp:690
↓ 79 callersFunctionas_position_independent_swizzle_tensor
3rd/cutlass/include/cute/pointer_flagged.hpp:99
↓ 79 callersMethodcompose
3rd/cutlass/include/cute/layout.hpp:190
↓ 79 callersMethodget_tile_offset
Obtains the threadblock offset (in units of threadblock-scoped tiles)
3rd/cutlass/include/cutlass/gemm/threadblock/threadblock_swizzle.h:131
↓ 78 callersMethodnon_const_ref
3rd/cutlass/include/cutlass/tensor_ref.h:244
↓ 77 callersFunctioncopy_if
3rd/cutlass/include/cute/algorithm/copy.hpp:47
↓ 76 callersMethodoffset
Computes the offset of an index from the origin of the tensor
3rd/cutlass/include/cutlass/tensor_ref.h:315
↓ 75 callersFunctionmake_identity_tensor
3rd/cutlass/include/cute/tensor_impl.hpp:495
↓ 74 callersFunctioncanonical_warp_idx_sync
Returns a warp-uniform value indicating the canonical warp index of the calling threads. Threads within the warp must be converged.
3rd/cutlass/include/cutlass/cutlass.h:126
next →1–100 of 10,542, ranked by callers