MCPcopy Create free account

hub / github.com/MeshInfra/WaferLLM / functions

Functions68 in github.com/MeshInfra/WaferLLM

⨍ Functions 68◇ Types & classes 4
↓ 13 callersFunctioncast_tensor_u32
(tensor)
Decode/launch_device.py:11
↓ 11 callersFunctioncast_tensor_u32
(tensor)
Prefill/launch_device.py:18
↓ 5 callersFunctionfloat_to_hex
(f)
SUMMA/host_common.py:17
↓ 5 callersFunctionfloat_to_hex
(f)
MeshGEMM/host_common.py:18
↓ 5 callersFunctionfloat_to_hex
(f)
MeshGEMV/host_common.py:16
↓ 5 callersFunctionfloat_to_hex
(f)
Prefill/launch_sim.py:11
↓ 5 callersFunctionfloat_to_hex
(f)
Prefill/launch_device.py:12
↓ 3 callersFunctionassignId
(pc, P)
MeshGEMM/host_common.py:26
↓ 3 callersFunctionassignId
(pc, P)
Prefill/launch_sim.py:17
↓ 3 callersFunctionassignId
(pc, P)
Prefill/launch_device.py:21
↓ 3 callersFunctionmake_u48
(words)
SUMMA/host_common.py:21
↓ 3 callersFunctionmake_u48
(words)
MeshGEMM/host_common.py:22
↓ 3 callersFunctionmake_u48
(words)
MeshGEMV/host_common.py:20
↓ 3 callersFunctionmake_u48
(words)
Prefill/launch_sim.py:14
↓ 3 callersFunctionmake_u48
(words)
Prefill/launch_device.py:15
↓ 2 callersFunctioncast_tensor_u32
(tensor)
SUMMA/launch_device.py:17
↓ 2 callersFunctioncast_tensor_u32
(tensor)
MeshGEMM/launch_device.py:17
↓ 2 callersFunctioncast_tensor_u32
(tensor)
MeshGEMV/launch_device.py:17
↓ 1 callersFunctionderive_params
(cj)
Prefill/compile.py:15
↓ 1 callersFunctionderive_params
(cj)
Decode/compile.py:15
↓ 1 callersFunctionmain
()
SUMMA/launch_sim.py:23
↓ 1 callersFunctionmain
()
SUMMA/launch_device.py:31
↓ 1 callersFunctionmain
()
SUMMA/compile.py:23
↓ 1 callersFunctionmain
()
MeshGEMM/launch_sim.py:23
↓ 1 callersFunctionmain
()
MeshGEMM/launch_device.py:31
↓ 1 callersFunctionmain
()
MeshGEMM/compile.py:25
↓ 1 callersFunctionmain
()
MeshGEMV/launch_sim.py:23
↓ 1 callersFunctionmain
()
MeshGEMV/launch_device.py:31
↓ 1 callersFunctionmain
()
MeshGEMV/compile.py:26
↓ 1 callersFunctionmain
()
Prefill/launch_sim.py:70
↓ 1 callersFunctionmain
()
Prefill/launch_device.py:74
↓ 1 callersFunctionmain
()
Prefill/compile.py:28
↓ 1 callersFunctionmain
()
Decode/launch_sim.py:29
↓ 1 callersFunctionmain
()
Decode/launch_device.py:34
↓ 1 callersFunctionmain
()
Decode/compile.py:34
↓ 1 callersFunctionparse_args
()
SUMMA/launch_sim.py:14
↓ 1 callersFunctionparse_args
()
SUMMA/launch_device.py:21
↓ 1 callersFunctionparse_args
()
SUMMA/compile.py:13
↓ 1 callersFunctionparse_args
()
MeshGEMM/launch_sim.py:14
↓ 1 callersFunctionparse_args
()
MeshGEMM/launch_device.py:21
↓ 1 callersFunctionparse_args
()
MeshGEMM/compile.py:15
↓ 1 callersFunctionparse_args
()
MeshGEMV/launch_sim.py:14
↓ 1 callersFunctionparse_args
()
MeshGEMV/launch_device.py:21
↓ 1 callersFunctionparse_args
()
MeshGEMV/compile.py:16
↓ 1 callersFunctionparse_args
()
Prefill/launch_sim.py:64
↓ 1 callersFunctionparse_args
()
Prefill/launch_device.py:68
↓ 1 callersFunctionparse_args
()
Decode/launch_sim.py:23
↓ 1 callersFunctionparse_args
()
Decode/launch_device.py:27
Method__init__
(self)
Prefill/launch_sim.py:55
Method__init__
(self)
Prefill/launch_device.py:59
Method__init__
(self)
Decode/launch_sim.py:12
Method__init__
(self)
Decode/launch_device.py:15
Functioncfg_name
Per-config suffix used for compile artifacts (out_<cfg>, artifact_<cfg>.json).
SUMMA/host_common.py:12
Functioncfg_name
Per-config suffix used for compile artifacts (out_<cfg>, artifact_<cfg>.json).
MeshGEMM/host_common.py:13
Functioncfg_name
Per-config suffix used for compile artifacts (out_<cfg>, artifact_<cfg>.json).
MeshGEMV/host_common.py:11
Functiondecode_timing
Decode the on-chip TSC timestamps into (mean_cycle, max_cycle).
SUMMA/host_common.py:49
Functiondecode_timing
Decode the on-chip TSC timestamps into (mean_cycle, max_cycle).
MeshGEMM/host_common.py:113
Functiondecode_timing
Decode the on-chip TSC timestamps into (mean_cycle, max_cycle).
MeshGEMV/host_common.py:39
Functionmake_inputs
(P, M, K, N)
SUMMA/host_common.py:25
Functionmake_inputs
Random X, W and the row-shuffled W the two-hop GEMM expects on device.
MeshGEMM/host_common.py:64
Functionmake_inputs
Random row-vector X (replicated across PE columns) and matrix W.
MeshGEMV/host_common.py:24
Functiontile_W
(tensor_W, P, Kt, Nt)
SUMMA/host_common.py:37
Functiontile_W
(tensor_W_offset, P, Kt, Nt)
MeshGEMM/host_common.py:101
Functiontile_W
(tensor_W, P, Mt, Nt)
MeshGEMV/host_common.py:33
Functiontile_X
(tensor_X, P, Mt, Kt)
SUMMA/host_common.py:31
Functiontile_X
(tensor_X, P, Mt, Kt)
MeshGEMM/host_common.py:95
Functionuntile_res
(res_1d_fp16, P, Mt, Nt)
SUMMA/host_common.py:43
Functionuntile_res
(res_1d_fp16, P, Mt, Nt)
MeshGEMM/host_common.py:107