MCPcopy Create free account
hub / github.com/UbiquitousLearning/mllm / CopyIn

Method CopyIn

mllm/backends/ascend/kernels/vec_add.cpp:40–47  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

38
39 private:
40 __aicore__ inline void CopyIn(int32_t progress) {
41 AscendC::LocalTensor<half> xLocal = inQueueX.AllocTensor<half>();
42 AscendC::LocalTensor<half> yLocal = inQueueY.AllocTensor<half>();
43 AscendC::DataCopy(xLocal, xGm[progress * TILE_LENGTH], TILE_LENGTH);
44 AscendC::DataCopy(yLocal, yGm[progress * TILE_LENGTH], TILE_LENGTH);
45 inQueueX.EnQue(xLocal);
46 inQueueY.EnQue(yLocal);
47 }
48 __aicore__ inline void Compute(int32_t progress) {
49 AscendC::LocalTensor<half> xLocal = inQueueX.DeQue<half>();
50 AscendC::LocalTensor<half> yLocal = inQueueY.DeQue<half>();

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected