Skip to content

Commit

Permalink
compilation error fix
Browse files Browse the repository at this point in the history
  • Loading branch information
v-Golubev committed Nov 15, 2023
1 parent 5c7ddb4 commit d9cbe2e
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -797,8 +797,8 @@ BrgemmEmitter::BrgemmEmitter(jit_generator* h, cpu_isa_t isa, const ExpressionPt
}
init_out_scheduling_params(output_desc);

auto brg0Prc = InferenceEngine::details::convertPrecision(brgemm_node->get_input_element_type(0));
auto brg1Prc = InferenceEngine::details::convertPrecision(brgemm_node->get_input_element_type(1));
const auto& brg0Prc = brgemm_node->get_input_element_type(0);
const auto& brg1Prc = brgemm_node->get_input_element_type(1);
bool brgWithAMX = brgemm_node->is_amx();

io_data_size = {brg0Prc.size(), brg1Prc.size()};
Expand Down Expand Up @@ -826,8 +826,8 @@ BrgemmEmitter::BrgemmEmitter(jit_generator* h, cpu_isa_t isa, const ExpressionPt
m_brgCtx.LDA = leading_dimensions[0];
m_brgCtx.LDB = leading_dimensions[1];
m_brgCtx.LDC = leading_dimensions[2];
m_brgCtx.dt_in0 = static_cast<dnnl_data_type_t>(DnnlExtensionUtils::IEPrecisionToDataType(brg0Prc));
m_brgCtx.dt_in1 = static_cast<dnnl_data_type_t>(DnnlExtensionUtils::IEPrecisionToDataType(brg1Prc));
m_brgCtx.dt_in0 = static_cast<dnnl_data_type_t>(DnnlExtensionUtils::ElementTypeToDataType(brg0Prc));
m_brgCtx.dt_in1 = static_cast<dnnl_data_type_t>(DnnlExtensionUtils::ElementTypeToDataType(brg1Prc));
m_brgCtx.beta = brgemm_node->get_beta();

initBrgemm(m_brgCtx, m_brgKernel, brgWithAMX);
Expand Down

0 comments on commit d9cbe2e

Please sign in to comment.