Skip to content

Commit edf4a4f

Browse files
authored
graph: backend: dnnl: enable block format in large partition kernel
1 parent f95a8c5 commit edf4a4f

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/graph/backend/dnnl/kernels/large_partition.cpp

+5
Original file line numberDiff line numberDiff line change
@@ -135,6 +135,11 @@ void larger_partition_kernel_t::setup_pipeline_stage1(
135135
void larger_partition_kernel_t::setup_pipeline_stage2(pass_pipeline_t &pipeline,
136136
memory_planner_t &mem_planner, bool enable_constant_cache) {
137137
pipeline.reset_visualize_arg(true, false);
138+
// do constant propagation here so that we can
139+
// prepare constant info for other optimizations.
140+
if (enable_constant_cache) {
141+
BACKEND_DNNL_ADD_PASS(pipeline, constant_propagation);
142+
}
138143
BACKEND_DNNL_ADD_PASS(pipeline, infer_shape);
139144
BACKEND_DNNL_ADD_PASS(pipeline, fuse_src_transpose_to_matmul);
140145
BACKEND_DNNL_ADD_PASS(pipeline, fuse_dst_transpose_to_matmul);

0 commit comments

Comments
 (0)