From 97f062b3cc64d31f2e9c023e9dc8faee80f5269b Mon Sep 17 00:00:00 2001 From: mbs-octoml Date: Tue, 12 Jul 2022 17:10:58 -0700 Subject: [PATCH] - fix relay.collage ffi prefix. --- src/relay/collage/cost_estimator.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/relay/collage/cost_estimator.h b/src/relay/collage/cost_estimator.h index fc02df95ab010..f433fd58401e7 100644 --- a/src/relay/collage/cost_estimator.h +++ b/src/relay/collage/cost_estimator.h @@ -54,7 +54,7 @@ class CostEstimatorNode : public Object { */ virtual Cost Estimate(const IRModule& mod, const Target& target, bool needs_tvm_tuning) const; - static constexpr const char* _type_key = "collage.CostEstimator"; + static constexpr const char* _type_key = "relay.collage.CostEstimator"; TVM_DECLARE_BASE_OBJECT_INFO(CostEstimatorNode, Object); }; @@ -77,7 +77,7 @@ class MockEstimatorNode : public CostEstimatorNode { public: Cost Estimate(const IRModule& mod, const Target& target, bool needs_tvm_tuning) const override; - static constexpr const char* _type_key = "collage.MockEstimator"; + static constexpr const char* _type_key = "relay.collage.MockEstimator"; TVM_DECLARE_FINAL_OBJECT_INFO(MockEstimatorNode, CostEstimatorNode); protected: