Siyuan Feng
e7af601636
Co-authored-by: Junru Shao <junrushao1994@gmail.com> Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com> Co-authored-by: Tristan Konolige <tristan.konolige@gmail.com>
106 lines
3.2 KiB
Python
106 lines
3.2 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
"""Helper functions in TVM Script Parser"""
|
|
|
|
from typing import Callable, List, Any, Optional, Tuple
|
|
|
|
import inspect
|
|
import synr
|
|
|
|
from tvm.ir import Span, SourceName
|
|
from tvm.error import DiagnosticError
|
|
|
|
|
|
def get_param_list(
|
|
func: Callable,
|
|
) -> Tuple[List[str], List[Tuple[str, Tuple[Any, ...]]], Optional[str]]:
|
|
"""Get the parameter list from definition of function"""
|
|
full_arg_spec: inspect.FullArgSpec = inspect.getfullargspec(func)
|
|
|
|
args: List[str]
|
|
defaults: Optional[Tuple[Any, ...]]
|
|
kwonlyargs: List[str]
|
|
args, defaults, kwonlyargs = (
|
|
full_arg_spec.args,
|
|
full_arg_spec.defaults,
|
|
full_arg_spec.kwonlyargs,
|
|
)
|
|
|
|
if defaults is None:
|
|
defaults = tuple()
|
|
|
|
if full_arg_spec.varkw is not None:
|
|
raise RuntimeError(
|
|
"TVM Script register error : variable keyword argument is not supported now"
|
|
)
|
|
|
|
if len(kwonlyargs) == 1 and kwonlyargs[0] == "span":
|
|
pass
|
|
elif not len(kwonlyargs) == 0:
|
|
raise RuntimeError("TVM Script register error : keyword only argument is not supported now")
|
|
|
|
pos_only: List[str] = list()
|
|
for arg in args[: len(args) - len(defaults)]:
|
|
if arg != "span":
|
|
pos_only.append(arg)
|
|
kwargs: List[Tuple[str, Tuple[Any, ...]]] = list()
|
|
for default, arg in zip(defaults, args[len(args) - len(defaults) :]):
|
|
if arg != "span":
|
|
kwargs.append((arg, default))
|
|
|
|
return pos_only, kwargs, full_arg_spec.varargs
|
|
|
|
|
|
def tvm_span_from_synr(span: synr.ast.Span) -> Span:
|
|
"""Convert a synr span to a TVM span"""
|
|
return Span(
|
|
SourceName(span.filename),
|
|
span.start_line,
|
|
span.end_line,
|
|
span.start_column,
|
|
span.end_column,
|
|
)
|
|
|
|
|
|
def synr_span_from_tvm(span: Span) -> synr.ast.Span:
|
|
"""Convert a TVM span to a synr span"""
|
|
return synr.ast.Span(
|
|
span.source_name.name,
|
|
span.line,
|
|
span.column,
|
|
span.end_line,
|
|
span.end_column,
|
|
)
|
|
|
|
|
|
def call_with_error_reporting(
|
|
report_error,
|
|
node_span,
|
|
func,
|
|
*args,
|
|
**kwargs,
|
|
):
|
|
"""Call function with exception handling and report error using node_span"""
|
|
try:
|
|
return func(*args, **kwargs)
|
|
except DiagnosticError:
|
|
raise
|
|
except Exception as err: # pylint: disable=broad-except
|
|
# printing last non-empty row of error message.
|
|
error_msg = list(filter(None, str(err).split("\n")))[-1]
|
|
report_error(error_msg, node_span)
|