From 8b6a7db03273f0c481ef0fea9f941e91b1ae32bd Mon Sep 17 00:00:00 2001 From: Dmitrii Zarukin Date: Thu, 16 Jan 2025 20:32:36 -0800 Subject: [PATCH] cpu: x64: ukernel: transform: add incorrect ldb check This fixes #2414 --- src/cpu/x64/brgemm/capi/brgemm_api.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/cpu/x64/brgemm/capi/brgemm_api.cpp b/src/cpu/x64/brgemm/capi/brgemm_api.cpp index 3dae3e3409c..e8eef66f8ec 100644 --- a/src/cpu/x64/brgemm/capi/brgemm_api.cpp +++ b/src/cpu/x64/brgemm/capi/brgemm_api.cpp @@ -680,6 +680,8 @@ status_t dnnl_transform_create(transform_t **transform, dim_t K, dim_t N, pack_type_t in_pack_type, dim_t in_ld, dim_t out_ld, data_type_t in_dt, data_type_t out_dt) { if (transform == nullptr) return status::invalid_arguments; + VCHECK_BRGEMM(utils::one_of(out_ld, 16, 32, 48, 64), + "Transform routine supports only \'out_ld\' of 16, 32, 48, or 64."); *transform = new transform_t(K, N, in_pack_type, in_ld, out_ld, in_dt, out_dt);