MCPcopy Create free account

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

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

MethodArguments
Default ctor
3rd/cutlass/include/cutlass/gemm/device/gemm_blockwise.h:665
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/gemm/device/gemm.h:660
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/gemm/device/gemm_sparse_with_visitor.h:191
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/gemm/device/gemm_complex.h:615
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/gemm/device/gemm_array.h:302
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/gemm/device/gemm_array.h:616
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/rank_k_universal.h:134
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/gemm_universal.h:150
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/gemm_sparse_universal_with_absmax.h:142
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/gemm/kernel/gemm_grouped.h:161
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/gemm_with_fused_epilogue.h:151
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/gemm_with_fused_epilogue.h:889
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/gemm_layernorm_mainloop_fusion.h:159
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/gemm_sparse_universal.h:300
MethodArguments
Default Constructor
3rd/cutlass/include/cutlass/gemm/kernel/gemm_universal_streamk.h:159
MethodArguments
Default constructor
3rd/cutlass/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized.hpp:155
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/gemv.h:156
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/gemv.h:429
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/gemm_with_absmax.h:143
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/rank_2k_universal.h:149
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/trmm_universal.h:135
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/gemm_planar_complex_array.h:146
MethodArguments
Default Constructor
3rd/cutlass/include/cutlass/gemm/kernel/gemm_streamk_with_fused_epilogue.h:168
MethodArguments
Default Constructor
3rd/cutlass/include/cutlass/gemm/kernel/gemm_streamk_with_fused_epilogue.h:1347
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/gemm_planar_complex.h:147
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/gemm_with_k_reduction.h:138
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/symm_universal.h:147
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/gemm/kernel/rank_2k_grouped.h:227
MethodArguments
3rd/cutlass/include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp:158
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/gemm/kernel/gemm_grouped_softmax_mainloop_fusion.h:164
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/conv/kernel/implicit_gemm_convolution.h:164
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/conv/kernel/implicit_gemm_convolution_with_absmax.h:173
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/conv/kernel/direct_convolution.h:314
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/conv/kernel/implicit_gemm_convolution_strided_dgrad.h:174
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/conv/kernel/implicit_gemm_convolution_fusion.h:167
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/conv/kernel/implicit_gemm_convolution_with_fused_epilogue.h:170
MethodArguments
Default ctor
3rd/cutlass/include/cutlass/reduction/device/reduce_split_k.h:91
MethodArguments
Methods
3rd/cutlass/include/cutlass/reduction/kernel/reduce_softmax_final.h:86
MethodArguments
Methods
3rd/cutlass/include/cutlass/epilogue/threadblock/epilogue_visitor_with_softmax.h:106
MethodArithmeticTuple
3rd/cutlass/include/cute/numeric/arithmetic_tuple.hpp:46
FunctionArithmeticTuple<T...> operator+
3rd/cutlass/include/cute/numeric/arithmetic_tuple.hpp:155
FunctionArithmeticTuple<T...> operator-
3rd/cutlass/include/cute/numeric/arithmetic_tuple.hpp:171
FunctionArithmeticTuple<U...> operator+
3rd/cutlass/include/cute/numeric/arithmetic_tuple.hpp:147
FunctionArithmeticTuple<U...> operator-
3rd/cutlass/include/cute/numeric/arithmetic_tuple.hpp:163
MethodArithmeticTupleIterator
3rd/cutlass/include/cute/numeric/arithmetic_tuple.hpp:191
FunctionArray<Element, 1> make_Array
3rd/cutlass/include/cutlass/array.h:521
FunctionArray<Element, 2> make_Array
3rd/cutlass/include/cutlass/array.h:527
FunctionArray<Element, 3> make_Array
3rd/cutlass/include/cutlass/array.h:533
FunctionArray<Element, 4> make_Array
3rd/cutlass/include/cutlass/array.h:539
MethodArray<Element, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_load_tma_warpspecialized.hpp:406
MethodArray<Element, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_load_tma_warpspecialized.hpp:564
MethodArray<Element, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_load_tma_warpspecialized.hpp:726
MethodArray<Element, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_load_tma_warpspecialized.hpp:891
FunctionArray<Element, N> ApplyArrayOperator
3rd/cutlass/include/cutlass/reduction/thread/reduction_operators.h:206
MethodArray<ElementAccumulator, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_load_tma_warpspecialized.hpp:68
MethodArray<ElementAccumulator, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_load_tma_warpspecialized.hpp:151
MethodArray<ElementAccumulator, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/threadblock/fusion/visitor_load.hpp:65
MethodArray<ElementCompute, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_load_tma_warpspecialized.hpp:1145
MethodArray<ElementOutput, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_compute_tma_warpspecialized.hpp:172
MethodArray<ElementOutput, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_compute_tma_warpspecialized.hpp:296
MethodArray<ElementOutput, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_compute_tma_warpspecialized.hpp:481
MethodArray<ElementOutput, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_compute_tma_warpspecialized.hpp:813
MethodArray<ElementOutput, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/fusion/sm100_visitor_compute_tma_warpspecialized.hpp:394
MethodArray<ElementOutput, FragmentSize> visit
3rd/cutlass/include/cutlass/epilogue/threadblock/fusion/visitor_compute.hpp:68
FunctionArray<T, N> fma
3rd/cutlass/include/cutlass/array.h:2824
MethodArray<T, N> operator()
3rd/cutlass/include/cutlass/epilogue/thread/activation.h:817
FunctionArray<T, N> operator*
3rd/cutlass/include/cutlass/array.h:2796
FunctionArray<T, N> operator+
3rd/cutlass/include/cutlass/array.h:2761
FunctionArray<T, N> operator-
3rd/cutlass/include/cutlass/array.h:2782
FunctionArray<T, N> operator/
3rd/cutlass/include/cutlass/array.h:2817
FunctionArray<float, 2> top_2_reduce
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_topk_softmax.hpp:84
FunctionArray<float, 2> top_2_reduce_scalar
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_topk_softmax.hpp:69
FunctionArray<float, 4> top_4_reduce
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_topk_softmax.hpp:126
FunctionArray<float, 4> top_4_reduce_scalar
3rd/cutlass/include/cutlass/epilogue/fusion/sm90_visitor_topk_softmax.hpp:101
FunctionArrayPlanarComplex<Element, N> make_ArrayPlanarComplex
3rd/cutlass/include/cutlass/array_planar_complex.h:79
MethodArrayProblemShape
3rd/cutlass/include/cutlass/gemm/group_array_problem_shape.hpp:138
MethodArrivalToken
3rd/cutlass/include/cutlass/pipeline/sm90_pipeline.hpp:120
MethodAutoFilter
3rd/cutlass/include/cute/algorithm/copy.hpp:279
MethodBarrier
src/communicator/communicator.cc:173
MethodBarrier
src/communicator/entry.cc:64
MethodBaseGrouped
Constructs the GEMM.
3rd/cutlass/include/cutlass/gemm/device/base_grouped.h:170
MethodBatchedGemmCoord
Default ctor
3rd/cutlass/include/cutlass/gemm_coord.h:277
MethodBroadcast
src/communicator/communicator.cc:75
FunctionCUTLASS_CONSTEXPR_IF_CXX17
3rd/cutlass/include/cutlass/exmy_base.h:215
MethodCUTLASS_CONSTEXPR_IF_CXX17
3rd/cutlass/include/cutlass/exmy_base.h:588
FunctionCUTLASS_HOST_DEVICE EllPredicatedTileAccessIterator
3rd/cutlass/include/cutlass/transform/threadblock/ell_predicated_tile_access_iterator.h:69
FunctionCUTLASS_HOST_DEVICE PredicatedScaleBiasVectorIterator
3rd/cutlass/include/cutlass/conv/threadblock/predicated_scale_bias_vector_iterator.h:74
FunctionCUTLASS_HOST_DEVICE PredicatedScaleBiasVectorIterator
3rd/cutlass/include/cutlass/transform/threadblock/predicated_scale_bias_vector_iterator.h:76
FunctionCUTLASS_HOST_DEVICE PredicatedTileAccessIterator
3rd/cutlass/include/cutlass/transform/threadblock/predicated_tile_access_iterator.h:329
FunctionCUTLASS_HOST_DEVICE PredicatedTileAccessIterator2dThreadTile
3rd/cutlass/include/cutlass/transform/threadblock/predicated_tile_access_iterator_2dthreadtile.h:79
FunctionCUTLASS_HOST_DEVICE PredicatedVectorAccessIterator
3rd/cutlass/include/cutlass/transform/threadblock/predicated_vector_access_iterator.h:87
FunctionCUTLASS_HOST_DEVICE bit_cast
3rd/cutlass/include/cutlass/platform/platform.h:550
MethodCallbacks
3rd/cutlass/include/cutlass/epilogue/threadblock/fusion/visitor_load.hpp:138
MethodCallbacks
3rd/cutlass/include/cutlass/epilogue/threadblock/fusion/visitor_2x.hpp:179
MethodCallbacks
3rd/cutlass/include/cutlass/epilogue/threadblock/fusion/visitor_2x.hpp:242
MethodChannel
src/communicator/channel.cc:24
MethodClose
src/communicator/listener.cc:132
MethodClusterBarrier
3rd/cutlass/include/cutlass/arch/barrier.h:352
MethodClusterTransactionBarrier
3rd/cutlass/include/cutlass/arch/barrier.h:548
FunctionCollectiveMma
3rd/cutlass/include/cutlass/gemm/collective/sm103_blockscaled_mma_array_warpspecialized.hpp:408
← previousnext →1,601–1,700 of 10,542, ranked by callers