aboutsummaryrefslogtreecommitdiff
path: root/src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h')
-rw-r--r--src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h6
1 files changed, 5 insertions, 1 deletions
diff --git a/src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h b/src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h
index 47548b2538..212fd79306 100644
--- a/src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h
+++ b/src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h
@@ -130,10 +130,14 @@ public:
{
return 3072;
}
- else
+ else if (platform.get_cpu_model() == CPUModel::A76)
{
return 4096;
}
+ else
+ {
+ return ICPPKernel::default_mws;
+ }
}
private: