未验证 提交 44cb3da3 编写于 作者: Y Yuanle Liu 提交者: GitHub

[Inference] fix pass_builder (#49595)

上级 8bbae468
...@@ -254,8 +254,6 @@ GpuPassStrategy::GpuPassStrategy() : PassStrategy({}) { ...@@ -254,8 +254,6 @@ GpuPassStrategy::GpuPassStrategy() : PassStrategy({}) {
"conv_elementwise_add_fuse_pass", // "conv_elementwise_add_fuse_pass", //
#endif // #endif //
"transpose_flatten_concat_fuse_pass", // "transpose_flatten_concat_fuse_pass", //
"constant_folding_pass", //
"auto_mixed_precision_pass", //
"conv2d_fusion_layout_transfer_pass", // "conv2d_fusion_layout_transfer_pass", //
"auto_mixed_precision_pass", // "auto_mixed_precision_pass", //
"inplace_op_var_pass", // should be the last pass. "inplace_op_var_pass", // should be the last pass.
......
...@@ -676,6 +676,8 @@ void BindAnalysisConfig(py::module *m) { ...@@ -676,6 +676,8 @@ void BindAnalysisConfig(py::module *m) {
py::arg("device_id") = 0, py::arg("device_id") = 0,
py::arg("precision_mode") = AnalysisConfig::Precision::kFloat32) py::arg("precision_mode") = AnalysisConfig::Precision::kFloat32)
.def("exp_enable_use_cutlass", &AnalysisConfig::Exp_EnableUseCutlass) .def("exp_enable_use_cutlass", &AnalysisConfig::Exp_EnableUseCutlass)
.def("exp_disable_mixed_precision_ops",
&AnalysisConfig::Exp_DisableMixedPrecisionOps)
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
.def("set_exec_stream", .def("set_exec_stream",
[](AnalysisConfig &self, phi::CUDAStream &stream) { [](AnalysisConfig &self, phi::CUDAStream &stream) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册