diff --git a/tf_adapter/kernels/geop_npu.cc b/tf_adapter/kernels/geop_npu.cc index 7ebfed1d060c18993e4de5f530afc0a4c3d03b83..485ecaf05d5ff4b43ecd6922822655467efd51aa 100644 --- a/tf_adapter/kernels/geop_npu.cc +++ b/tf_adapter/kernels/geop_npu.cc @@ -918,6 +918,7 @@ void GeOp::ComputeAsync(OpKernelContext *ctx, DoneCallback done) { if (!recompute_mode_.empty()) { graph_options_["ge.recompute"] = recompute_mode_; } + if (!deploy_inject_config_.empty()) { graph_options_["ge.exec.clusterSpec"] = deploy_inject_config_; }