MCPcopy Create free account

hub / github.com/KnowingNothing/MatmulTutorial / functions

Functions541 in github.com/KnowingNothing/MatmulTutorial

⨍ Functions 541◇ Types & classes 125
Functionlinear_attention_fwd_kernel
( realQ, imageQ, realK, imageK, realV, imageV, realO, imageO, # realP, imageP, real_sm_scale, image_sm
examples/attention/triton/fused_linear_attention_complex_tile_k.py:19
Methodmake_pipeline_state
include/barrier.h:221
Functionmake_tma_copy_desc
include/tma.h:154
Functionmatmul_kernel
Kernel for computing the matmul C = A x B. A has shape (M, K), B has shape (K, N) and C has shape (M, N)
examples/matmul/triton/triton_matmul.py:180
Functionmatmul_kernel_tma_persistent
( a_desc_ptr, b_desc_ptr, c_desc_ptr, # M, N, K, # BLOCK_SIZE_M: tl.constexpr,
examples/matmul/triton/triton_matmul_sm90.py:71
Methodnum_elements
(self)
cutlass.py/tiling.py:62
Methodnum_elements_ge
We can conclude if number of elements larger than a given value. Because all dimensions should be > 0.
cutlass.py/tiling.py:75
Methodnum_elements_gt
We can conclude if number of elements larger than a given value. Because all dimensions should be > 0.
cutlass.py/tiling.py:68
Methodoperator()
include/common.h:99
Methodoperator()
include/common.h:489
Methodoperator++
include/barrier.h:185
Methodoperator==
include/pipeline.h:24
Methodpossibly_truncate
(x, y)
cutlass.py/tile_scheduler.py:145
Functionprefetch_tma_descriptor
include/descriptor.h:90
Functionproducer_commit
include/pipeline.h:183
Functionproducer_tail
Prevents early exit of producer blocks in Cluster. This should be called once before kernel exits.
include/pipeline.h:189
Functionrandom_fill
include/common.h:283
Methodreduction_subtile_idx
(self)
cutlass.py/tile_scheduler.py:255
Functionreference_cpu_gemm
include/reference.h:6
Functionreference_gpu_gemm
include/reference.h:79
Functionreference_gpu_gemm_kernel
include/reference.h:51
Methodreset_bytes
include/barrier.h:428
Methodset_threadIdx
(self, x, y, z)
cutlass.py/hw_info.py:39
Methodsetup_context
(ctx, inputs, output)
examples/attention/triton/model.py:67
Methodstart
Start the timer for a given stream (defaults to the default stream)
examples/matmul/cutlass/helper.h:88
Methodstop
Stop the timer
examples/matmul/cutlass/helper.h:95
Functionsupports_tma
()
examples/matmul/triton/triton_matmul_sm90.py:33
Methodtest_wait
include/barrier.h:40
MethodthreadRange
(self)
util/simulator/kernel.py:177
Functionto_CUtensorMapDataType
include/tma.h:97
Functionuint16_t
()
cutlass.py/dtype.py:187
Functionuint32_t
()
cutlass.py/dtype.py:191
Functionuint4_t
()
cutlass.py/dtype.py:179
Functionuint64_t
Decay to a uint64_t
include/descriptor.h:70
Functionuint64_t
()
cutlass.py/dtype.py:195
Functionuint8_t
()
cutlass.py/dtype.py:183
Methodvalue_type
include/common.h:98
Methodwait
Wait on a stage to be unlocked
include/barrier.h:280
Functionwarp_id_in_warp_group
Returns the warp id in warp group
include/common.h:447
Method~GpuTimer
Destructor
examples/matmul/cutlass/helper.h:81
Method~OrderedSequenceBarrier
include/barrier.h:257
← previous501–541 of 541, ranked by callers