单核M1 CPU上实现FP32 1.5 TFlops算力？这是一份代码指南

void reference_16x16xK (float *A, float *B, float *C, uint64_t K) {  for (uint32_t m = 0; m < 16; ++m) {    for (uint32_t n = 0; n < 16; ++n) {      C [n * 16 + m] = 0;      for (uint32_t k = 0; k < K; ++k) {        C [n * 16 + m] += A [k * 16 + m] * B [k * 16 + n];      }    }  }}

/only set for k == 0uint64_t reset_z = 1ull << 27;for (uint32_t k = 0; k < K; ++k) {  uint64_t idx = k % 4;  // 64 bytes = 16 floats  AMX_LDX ((uint64_t) A + k * 64);  AMX_LDY ((uint64_t) B + k * 64);    //now we do 4 indepedent outer products (avoiding pipeline hazards)  AMX_FMA32 (reset_z);  reset_z = 0;}

for (uint64_t i = 0; i < 16; ++i) {  const uint64_t z_register = (i * 4ull) << 56;  AMX_STZ (z_register | (uint64_t) C + i * 64);}

void mm32x32xK (float* A, float* B, float* C, uint64_t K) {
  //flag to load/store 128 bytes  const uint64_t load_store_2 = 1ull << 62;  const uint64_t load_store_width = 128; //in bytes
  //only set for k == 0  uint64_t reset_z = 1ull << 27;

  for (uint32_t k = 0; k < K; ++k) {    uint64_t idx = k % 4;    //load to X, Y (skipping every other index because we're loading 128 bytes)    AMX_LDX (load_store_2 | (idx * 2) << 56 | (uint64_t) A + k * load_store_width);    AMX_LDY (load_store_2 | (idx * 2) << 56 | (uint64_t) B + k * load_store_width);
    //offset into X and Y registers is byte-wise    const uint64_t offset = idx * load_store_width;
    //now we do 4 indepedent outer products (avoiding pipeline hazards)    AMX_FMA32 (reset_z | (0ull << 20) | ((offset +  0ull) << 10) | ((offset +  0ull) << 0));    AMX_FMA32 (reset_z | (1ull << 20) | ((offset + 64ull) << 10) | ((offset +  0ull) << 0));    AMX_FMA32 (reset_z | (2ull << 20) | ((offset +  0ull) << 10) | ((offset + 64ull) << 0));    AMX_FMA32 (reset_z | (3ull << 20) | ((offset + 64ull) << 10) | ((offset + 64ull) << 0));    reset_z = 0;  }
  for (uint64_t i = 0; i < 16; ++i) {    //store interleaved    AMX_STZ (load_store_2 | ((i * 4ull + 0) << 56) | (uint64_t) C + i * load_store_width);    AMX_STZ (load_store_2 | ((i * 4ull + 2) << 56) | (uint64_t) C + (16 + i) * load_store_width);  }}