From 3b0983e4ef4c942de84bed5483bd73db741f953f Mon Sep 17 00:00:00 2001 From: Junru Shao Date: Fri, 3 Dec 2021 13:18:49 -0800 Subject: [PATCH] [Bugfix] BuilderInput with default params (#532) --- python/tvm/meta_schedule/builder/builder.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/python/tvm/meta_schedule/builder/builder.py b/python/tvm/meta_schedule/builder/builder.py index 43c7f2ea2d..7278c458a6 100644 --- a/python/tvm/meta_schedule/builder/builder.py +++ b/python/tvm/meta_schedule/builder/builder.py @@ -45,7 +45,12 @@ class BuilderInput(Object): target: Target params: Optional[Dict[str, NDArray]] - def __init__(self, mod: IRModule, target: Target, params: Optional[Dict[str, NDArray]]) -> None: + def __init__( + self, + mod: IRModule, + target: Target, + params: Optional[Dict[str, NDArray]] = None, + ) -> None: """Constructor. Parameters