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

Function tile_X

MeshGEMM/host_common.py:95–98  ·  view source on GitHub ↗
(tensor_X, P, Mt, Kt)

Source from the content-addressed store, hash-verified

93
94
95def tile_X(tensor_X, P, Mt, Kt):
96 X1 = tensor_X.reshape(P, Mt, P, Kt)
97 X2 = X1.transpose(0, 2, 3, 1)
98 return X2.reshape(P, P, Mt * Kt).ravel()
99
100
101def tile_W(tensor_W_offset, P, Kt, Nt):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected