Skip to content

Commit 53de57f

Browse files
committed
cpu: aarch64: Update the new files
1 parent a892566 commit 53de57f

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

src/cpu/aarch64/jit_brgemm_conv_bwd.cpp

+4-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,9 @@
2020
#include "common/type_helpers.hpp"
2121
#include "common/utils.hpp"
2222

23+
#include "cpu/aarch64/jit_brgemm_1x1_conv.hpp"
2324
#include "cpu/aarch64/jit_brgemm_conv_bwd.hpp"
25+
#include "cpu/cpu_convolution_pd.hpp"
2426

2527
namespace dnnl {
2628
namespace impl {
@@ -130,7 +132,8 @@ status_t brgemm_convolution_bwd_t<isa>::pd_t::init(engine_t *engine) {
130132
break; // non-1x1 implementation found
131133
}
132134
}
133-
if (it == it.end()) { return status::unimplemented; }
135+
136+
VDISPATCH_CONV(it != it.end(), "Implementation wasn't found");
134137

135138
if (weights_md_.format_kind == format_kind::any)
136139
CHECK(weights_axes_permutation(

src/cpu/aarch64/jit_brgemm_conv_bwd.hpp

+2-5
Original file line numberDiff line numberDiff line change
@@ -14,18 +14,15 @@
1414
* limitations under the License.
1515
*******************************************************************************/
1616

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
1919

2020
#include "common/c_types_map.hpp"
2121
#include "common/dnnl_thread.hpp"
2222
#include "common/memory_tracking.hpp"
2323
#include "common/primitive.hpp"
2424
#include "common/utils.hpp"
2525

26-
#include "cpu/cpu_convolution_pd.hpp"
27-
28-
#include "cpu/aarch64/jit_brgemm_1x1_conv.hpp"
2926
#include "cpu/aarch64/jit_brgemm_conv.hpp"
3027

3128
namespace dnnl {

0 commit comments

Comments
 (0)