[BUGFIX] Fix remove Cast fuse (#21086)

* Fix remove Cast fuse

* Fix Reset()

* Fix review

* Fix sanity

src/operator/subgraph/dnnl/dnnl_remove_casts_property.h

@@ -95,7 +95,7 @@ class SgDNNLRemoveCastsSelector : public SubgraphSelectorV2 {
  }
 
  void Reset() override {
- status_ = kFail;
+ status_ = kExpand;
  castDtype = -1;
  }
 };
@@ -105,7 +105,7 @@ class SgDNNLRemoveCastsProperty : public SubgraphProperty {
  SgDNNLRemoveCastsProperty() {}
 
  static SubgraphPropertyPtr Create() {
- static const std::string& name = "Remove casts optimization pass";
+ static const std::string& name = "Remove Casts optimization pass";
  auto property = std::make_shared<SgDNNLRemoveCastsProperty>();
  property->SetAttr<std::string>("property_name", name);
  property->SetAttr<bool>("inference_only", true);
@@ -137,6 +137,14 @@ class SgDNNLRemoveCastsProperty : public SubgraphProperty {
  auto selector = std::make_shared<SgDNNLRemoveCastsSelector>();
  return selector;
  }
+
+ void ConnectSubgraphOutputs(const nnvm::ObjectPtr subgraph_node,
+ std::vector<nnvm::NodeEntry*>* output_entries) const override {
+ // Connect all extern output entries to output[0]
+ for (size_t i = 0; i < output_entries->size(); ++i) {
+ *output_entries->at(i) = nnvm::NodeEntry{subgraph_node, 0, 0};
+ }
+ }
 };
 
 } // namespace op