We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f95a8c5 commit edf4a4fCopy full SHA for edf4a4f
src/graph/backend/dnnl/kernels/large_partition.cpp
@@ -135,6 +135,11 @@ void larger_partition_kernel_t::setup_pipeline_stage1(
135
void larger_partition_kernel_t::setup_pipeline_stage2(pass_pipeline_t &pipeline,
136
memory_planner_t &mem_planner, bool enable_constant_cache) {
137
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
+ }
143
BACKEND_DNNL_ADD_PASS(pipeline, infer_shape);
144
BACKEND_DNNL_ADD_PASS(pipeline, fuse_src_transpose_to_matmul);
145
BACKEND_DNNL_ADD_PASS(pipeline, fuse_dst_transpose_to_matmul);
0 commit comments