Unverified Commit e883cc8a authored by PanZezhong1725's avatar PanZezhong1725 Committed by GitHub
Browse files

Merge pull request #363 from InfiniTensor/issue/362

issue/362 - allowed bfloat16_t for cambricon gemm
parents 82e43d35 04d1f27c
...@@ -67,9 +67,7 @@ infiniStatus_t Descriptor::create( ...@@ -67,9 +67,7 @@ infiniStatus_t Descriptor::create(
auto handle = reinterpret_cast<device::bang::cambricon::Handle *>(handle_); auto handle = reinterpret_cast<device::bang::cambricon::Handle *>(handle_);
auto dtype = c_desc->dtype(); auto dtype = c_desc->dtype();
if (dtype != INFINI_DTYPE_F16 && dtype != INFINI_DTYPE_F32) { CHECK_DTYPE(dtype, INFINI_DTYPE_F16, INFINI_DTYPE_BF16, INFINI_DTYPE_F32);
return INFINI_STATUS_BAD_TENSOR_DTYPE;
}
auto result = MatmulInfo::create(c_desc, a_desc, b_desc, MatrixLayout::ROW_MAJOR); auto result = MatmulInfo::create(c_desc, a_desc, b_desc, MatrixLayout::ROW_MAJOR);
CHECK_RESULT(result); CHECK_RESULT(result);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment