File tree 2 files changed +6
-6
lines changed
2 files changed +6
-6
lines changed Original file line number Diff line number Diff line change 20
20
#include " common/type_helpers.hpp"
21
21
#include " common/utils.hpp"
22
22
23
+ #include " cpu/aarch64/jit_brgemm_1x1_conv.hpp"
23
24
#include " cpu/aarch64/jit_brgemm_conv_bwd.hpp"
25
+ #include " cpu/cpu_convolution_pd.hpp"
24
26
25
27
namespace dnnl {
26
28
namespace impl {
@@ -130,7 +132,8 @@ status_t brgemm_convolution_bwd_t<isa>::pd_t::init(engine_t *engine) {
130
132
break ; // non-1x1 implementation found
131
133
}
132
134
}
133
- if (it == it.end ()) { return status::unimplemented; }
135
+
136
+ VDISPATCH_CONV (it != it.end (), " Implementation wasn't found" );
134
137
135
138
if (weights_md_.format_kind == format_kind::any)
136
139
CHECK (weights_axes_permutation (
Original file line number Diff line number Diff line change 14
14
* limitations under the License.
15
15
*******************************************************************************/
16
16
17
- #ifndef CPU_X64_JIT_BRGEMM_CONV_BWD_HPP
18
- #define CPU_X64_JIT_BRGEMM_CONV_BWD_HPP
17
+ #ifndef CPU_AARCH64_JIT_BRGEMM_CONV_BWD_HPP
18
+ #define CPU_AARCH64_JIT_BRGEMM_CONV_BWD_HPP
19
19
20
20
#include " common/c_types_map.hpp"
21
21
#include " common/dnnl_thread.hpp"
22
22
#include " common/memory_tracking.hpp"
23
23
#include " common/primitive.hpp"
24
24
#include " common/utils.hpp"
25
25
26
- #include " cpu/cpu_convolution_pd.hpp"
27
-
28
- #include " cpu/aarch64/jit_brgemm_1x1_conv.hpp"
29
26
#include " cpu/aarch64/jit_brgemm_conv.hpp"
30
27
31
28
namespace dnnl {
You can’t perform that action at this time.
0 commit comments