Skip to content

Commit

Permalink
cpu: ref_deconv: add limitation on supported bias data type
Browse files Browse the repository at this point in the history
  • Loading branch information
dzarukin committed Jan 9, 2025
1 parent c1228bd commit 3100fd1
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion src/cpu/ref_deconvolution.hpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*******************************************************************************
* Copyright 2018-2024 Intel Corporation
* Copyright 2018-2025 Intel Corporation
* Copyright 2022 Arm Ltd. and affiliates
*
* Licensed under the Apache License, Version 2.0 (the "License");
Expand Down Expand Up @@ -173,6 +173,16 @@ struct ref_deconvolution_fwd_t : public primitive_t {
alg_kind::deconvolution_direct,
alg_kind::deconvolution_winograd),
VERBOSE_BAD_ALGORITHM);
// This implementation will check data types requirements through
// an underlying convolution implementation, however, convolution
// might be called without bias, thus, need to check bias data type
// if it was requested.
if (with_bias()) {
const auto bia_type = weights_md(1)->data_type;
VDISPATCH_DECONVOLUTION(utils::one_of(bia_type, f32, bf16, f16,
f8_e5m2, f8_e4m3),
VERBOSE_UNSUPPORTED_DT);
}
VDISPATCH_DECONVOLUTION(attr()->has_default_values(skip_mask),
VERBOSE_UNSUPPORTED_ATTR);
VDISPATCH_DECONVOLUTION(
Expand Down

0 comments on commit 3100fd1

Please sign in to comment.