MCPcopy Create free account
hub / github.com/MeshInfra/WaferLLM / tile_W

Function tile_W

MeshGEMM/host_common.py:101–104  ·  view source on GitHub ↗
(tensor_W_offset, P, Kt, Nt)

Source from the content-addressed store, hash-verified

99
100
101def tile_W(tensor_W_offset, P, Kt, Nt):
102 W1 = tensor_W_offset.reshape(P, Kt, P, Nt)
103 W2 = W1.transpose(0, 2, 1, 3)
104 return W2.reshape(P, P, Kt * Nt).ravel()
105
106
107def untile_res(res_1d_fp16, P, Mt, Nt):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected