File tree 2 files changed +4
-1
lines changed
cinn/hlir/dialect/operator/transforms
2 files changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -26,6 +26,9 @@ void SpecifyInputDynamicDim(
26
26
bool to_append = false );
27
27
void SpecifyInputDynamicDimFromFile (pir::Program* program,
28
28
std::string filepath);
29
+
30
+ void SpecifyInputDynamicDimFromTriplet (
31
+ pir::Program* program, const std::vector<NamedTriplet>& constraints);
29
32
} // namespace ir
30
33
} // namespace dialect
31
34
} // namespace cinn
Original file line number Diff line number Diff line change @@ -3290,7 +3290,7 @@ void BindShapeConstraintIRAnalysis(pybind11::module *m) {
3290
3290
});
3291
3291
3292
3292
m->def (" specify_input_dynamic_dim_from_triplet" ,
3293
- &SpecifyInputDynamicDimFromTriplet);
3293
+ &::cinn::dialect::ir:: SpecifyInputDynamicDimFromTriplet);
3294
3294
3295
3295
py::class_<pir::ShapeConstraintIRAnalysis,
3296
3296
std::shared_ptr<pir::ShapeConstraintIRAnalysis>>
You can’t perform that action at this time.
0 commit comments