fix bug

cinn/hlir/op/contrib/squeeze.cc

@@ -188,21 +188,6 @@ std::vector<Type> InferDtypeForSqueeze(const std::vector<Type> &inputs_type, con
   return res;
 }
 
-std::vector<std::vector<std::string>> InferLayoutForSqueeze(const std::vector<framework::shape_t> &input_shapes,
-                                                            const std::vector<std::string> &input_layouts,
-                                                            const framework::NodeAttr &attrs,
-                                                            const Target &target) {
-  CHECK_EQ(input_shapes.size(), 1U) << "The input's shape size is not 1! Please check again.";
-  CHECK_EQ(input_layouts.size(), 1U) << "The input's layout size is not 1! Please check again.";
-  std::vector<std::string> new_input_layouts = input_layouts;
-  if (input_shapes[0].size() > 4) {
-    // alter input layout back
-    new_input_layouts[0] = "NCHW";
-    VLOG(3) << "alter input layout from " << input_layouts[0] << " to " << new_input_layouts[0];
-  }
-  return {new_input_layouts, new_input_layouts};
-}
-
 }  // namespace op
 }  // namespace hlir
 }  // namespace cinn
@@ -215,9 +200,6 @@ CINN_REGISTER_HELPER(squeeze_ops) {
       .set_attr<cinn::hlir::framework::StrategyFunction>("CINNStrategy", cinn::hlir::op::StrategyForSqueeze)
       .set_attr("infershape", MakeOpFunction(cinn::hlir::op::InferShapeForSqueeze))
       .set_attr("inferdtype", MakeOpFunction(cinn::hlir::op::InferDtypeForSqueeze))
-#ifndef CINN_WITH_CUDA
-      .set_attr("inferlayout", MakeOpFunction(cinn::hlir::op::InferLayoutForSqueeze))
-#endif
       .set_support_level(4);
 
   return true;

python/tests/ops/test_squeeze_op.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python3
-
 # Copyright (c) 2022 CINN Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");