[CPU] Add CMakeLists.txt for sgl-kernel (#6115)

This commit is contained in:
blzheng
2025-05-14 06:30:37 +08:00
committed by GitHub
parent 16267d4fa7
commit 0f75b907c6
6 changed files with 132 additions and 3 deletions

View File

@@ -1,7 +1,8 @@
#pragma once
#include <ATen/native/CPUBlas.h>
#include "common.h"
// amx-bf16
#define TILE_M 16
#define TILE_N 16