mindspore 2.2.11__cp37-cp37m-manylinux1_x86_64.whl → 2.3.0rc1__cp37-cp37m-manylinux1_x86_64.whl

mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/build_tbe_kernel.py

@@ -0,0 +1,529 @@
+# Copyright 2021-2023 Huawei Technologies Co., Ltd
+#
+# Licensed 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.
+
+"""build tbe kernel"""
+
+import os
+import json
+import functools
+from tbe_topi import OpPattern, get_op_reg_info
+from te import tvm
+from te.utils import shape_util
+from te.platform.cce_conf import te_set_version
+from tbe.common.buildcfg import build_config
+from tbe.dsl import auto_schedule
+from tbe.dsl import build as tbe_build
+import tbe.common.context.op_context as op_context
+from impl.dynamic.add import _add_check_format, _infer_shape
+
+SHAPE = "shape"
+FORMAT = "format"
+DATA_TYPE = "data_type"
+NEW_SHAPE = "new_shape"
+ORI_SHAPE = "ori_shape"
+ORI_FORMAT = "ori_format"
+DST_TYPE = "dst_type"
+DST_ORI_SHAPE = "dst_ori_shape"
+INPUT_DESC = "input_desc"
+OUTPUT_DESC = "output_desc"
+ENABLE_VECTOR_2X = "enable_vector_2x"
+ENABLE_GROUP_INPLACE = "enable_group_inplace"
+
+
+def get_arch_name(op_json):
+    """Get the arch name from op json."""
+    target_info = op_json.get("target_info")
+    if isinstance(target_info, dict):
+        arch_name = target_info.get("arch")
+        if arch_name:
+            return arch_name
+    return "Ascend910A"
+
+
+def initialize(kernel_meta_parent_dir, arch_name):
+    """Initialize the TBE compile environment."""
+    os.environ["CONTEXT_MODELCOMPILING"] = "TRUE"
+    core_type = "VectorCore" if arch_name != "Ascend910A" else ""
+    # socVersion, coreType, coreNum, l1Fusion, l2Mode, l2Fusion
+    soc_info = [arch_name, core_type, "", "false", "2", "false",
+                {"op_impl_mode": "",
+                 "op_debug_level": "3",
+                 "op_impl_mode_list": [],
+                 "op_debug_dir": kernel_meta_parent_dir,
+                 "vector_fp_ceiling": "",
+                 "mdl_bank_path": "",
+                 "op_bank_path": ""}]
+    if not te_set_version(*soc_info):
+        raise RuntimeError("Set version failed")
+
+
+def update_config(config, op_names):
+    """Update build config."""
+    bool_storage_as_1bit_oplist = ["Asinh", "Atanh", "Acosh", "Asin", "Atan2", "Acos", "Pow", "Elu", "Select"]
+    change_type_dict = {"MatMul": (True, False),
+                        "BatchMatMul": (True, False)}
+    config["bool_storage_as_1bit"] = True
+    config[ENABLE_GROUP_INPLACE] = False
+    config[ENABLE_VECTOR_2X] = True
+    for op in op_names:
+        if op in bool_storage_as_1bit_oplist:
+            config["bool_storage_as_1bit"] = False
+        enable_group_inplace, enable_vector_2x = change_type_dict.get(op, (False, True))
+        config[ENABLE_GROUP_INPLACE] = config[ENABLE_GROUP_INPLACE] or enable_group_inplace
+        config[ENABLE_VECTOR_2X] = config[ENABLE_VECTOR_2X] and enable_vector_2x
+
+
+def add_new_shape(names, shapes, new_shapes, inputs):
+    """Add new shape for input tensors."""
+    if not isinstance(names, list):
+        names = [names]
+        shapes = [shapes]
+        new_shapes = [new_shapes]
+    for i, name in enumerate(names):
+        if shapes[i] == new_shapes[i]:
+            continue
+        if name not in inputs:
+            continue
+        if NEW_SHAPE not in inputs[name]:
+            inputs[name][NEW_SHAPE] = new_shapes[i]
+        elif new_shapes[i] != inputs[name][NEW_SHAPE]:
+            raise RuntimeError("Find different new_shape {} and {} for {}"
+                               .format(inputs[name][NEW_SHAPE], new_shapes[i], name))
+
+
+class TransShape:
+    """TransShape"""
+
+    @classmethod
+    def trans_elemwise_shape(cls, op_inputs, inputs):
+        """deal with elemwise shape."""
+        names = []
+        shapes = []
+        ori_shapes = []
+        formats = []
+        ori_formats = []
+        for k, v in op_inputs.items():
+            # scalar not included
+            if v.get("value") is not None:
+                continue
+            names.append(k)
+            shapes.append(v[SHAPE])
+            ori_shapes.append(v[ORI_SHAPE] if v.get(ORI_SHAPE) else None)
+            formats.append(v[FORMAT])
+            ori_formats.append(v[ORI_FORMAT])
+        if len(shapes) == 2 and len(shapes[0]) != len(shapes[1]):
+            format_pattern = _add_check_format({SHAPE: shapes[0], FORMAT: formats[0]},
+                                               {SHAPE: shapes[1], FORMAT: formats[1]})
+            ori_shape0 = ori_shapes[0] if ori_shapes[0] is not None else infer_ori_shape(
+                shapes[0], formats[0], ori_formats[0])
+            ori_shape1 = ori_shapes[1] if ori_shapes[1] is not None else infer_ori_shape(
+                shapes[1], formats[1], ori_formats[1])
+            new_shapes = [None, None]
+            new_shapes[0], new_shapes[1] = _infer_shape(format_pattern,
+                                                        {SHAPE: shapes[0], ORI_SHAPE: ori_shape0},
+                                                        {SHAPE: shapes[1], ORI_SHAPE: ori_shape1})
+            new_shapes[0], new_shapes[1], _ = shape_util.broadcast_shapes(new_shapes[0], new_shapes[1],
+                                                                          param_name_input1="input0",
+                                                                          param_name_input2="input1")
+            add_new_shape(names, shapes, new_shapes, inputs)
+
+    @classmethod
+    def trans_batch_matmul_shape(cls, op_inputs, inputs):
+        """deal with batch_matmul."""
+        for k, v in op_inputs.items():
+            # batch dimension of BatchMatMul must be fused to 1D
+            shape = v[SHAPE]
+            if len(shape) > 5:
+                new_shape = [functools.reduce(lambda x, y: x * y, shape[:-4])] + shape[-4:]
+                add_new_shape(k, shape, new_shape, inputs)
+
+    @classmethod
+    def run(cls, op_name, pattern, op_inputs, inputs):
+        """entry function."""
+        if pattern == OpPattern.ELEMWISE:
+            TransShape.trans_elemwise_shape(op_inputs, inputs)
+        elif op_name == "BatchMatMul":
+            TransShape.trans_batch_matmul_shape(op_inputs, inputs)
+
+
+def infer_ori_shape(shape, cur_format, ori_format):
+    """Given current format and shape, infer the shape with ori_format."""
+
+    def _shape_error(current_shape, current_format):
+        raise ValueError("Invalid shape {} for format {}".format(current_shape, current_format))
+
+    if cur_format == ori_format:
+        return shape
+    default_formats = ["DefaultFormat", "ND", "NCHW"]
+
+    if cur_format in default_formats and ori_format in default_formats:
+        return shape
+
+    if cur_format == "FRACTAL_NZ" and ori_format in default_formats:
+        dims = len(shape)
+        if dims < 4:
+            _shape_error(shape, cur_format)
+        ori_shape = shape[:dims - 4]
+        m = shape[-3] * shape[-2]
+        n = shape[-4] * shape[-1]
+        ori_shape.append(m)
+        ori_shape.append(n)
+        return ori_shape
+
+    if cur_format == "NC1HWC0" and ori_format in default_formats:
+        if len(shape) != 5:
+            _shape_error(shape, cur_format)
+        ori_shape = [shape[0], shape[1] * shape[4], shape[2], shape[3]]
+        return ori_shape
+
+    if cur_format == "NHWC" and ori_format in default_formats:
+        if len(shape) != 4:
+            _shape_error(shape, cur_format)
+        ori_shape = [shape[0], shape[3], shape[1], shape[2]]
+        return ori_shape
+
+    raise ValueError("Not support infer shape from format {} to {}".format(cur_format, ori_format))
+
+
+def get_inputs_name(input_desc):
+    """Get input tensor names."""
+    inputs_name = []
+    for desc in input_desc:
+        for item in desc:
+            inputs_name.append(item["tensor_name"])
+    return inputs_name
+
+
+def get_outputs_info(output_desc):
+    """Get output tensor names and inplace tensor names."""
+    outputs_name = []
+    inplace_names = {}
+    for item in output_desc:
+        out_name = item["tensor_name"]
+        outputs_name.append(out_name)
+        if isinstance(item.get("inplace_to"), str):
+            inplace_names[out_name] = item["inplace_to"]
+    return outputs_name, inplace_names
+
+
+def get_all_op_name(op_desc):
+    """Get op names."""
+    op_names = []
+    for op in op_desc:
+        op_names.append(op["name"])
+    return op_names
+
+
+def get_input_desc(input_desc):
+    """Get input desc."""
+    res = {}
+    for desc in input_desc:
+        for item in desc:
+            item[SHAPE] = [1] if not item[SHAPE] else item[SHAPE]
+            res[item["tensor_name"]] = item
+    return res
+
+
+def get_inputs_tensor(input_desc, all_tensors):
+    """Get input placeholders."""
+    inputs = []
+    for desc in input_desc:
+        for item in desc:
+            name = item["tensor_name"]
+            if item.get("value") is not None:
+                # const value
+                all_tensors[name] = tvm.const(item["value"], item[DATA_TYPE])
+            if all_tensors.get(name) is None:
+                raise ValueError("Tensor [{}] not found.".format(name))
+            inputs.append(all_tensors[name])
+    return inputs
+
+
+def get_attr_dict(attr_desc):
+    """Parse attr_desc to dict."""
+    attrs = {}
+    if not isinstance(attr_desc, list):
+        return attrs
+    for attr in attr_desc:
+        attrs[attr["name"]] = attr["value"]
+    return attrs
+
+
+def get_op_attrs(op, fusion_op_name):
+    """Get op attrs."""
+    op_name = op["name"]
+    op_attrs = get_attr_dict(op.get("attr"))
+    if op_name == "BatchMatMul":
+        op_attrs[DST_TYPE] = op[OUTPUT_DESC][0][DATA_TYPE]
+        op_attrs[DST_ORI_SHAPE] = op[OUTPUT_DESC][0].get(ORI_SHAPE)
+        if op_attrs.get(DST_ORI_SHAPE) is None:
+            op_attrs[DST_ORI_SHAPE] = infer_ori_shape(op[OUTPUT_DESC][0][SHAPE],
+                                                      op[OUTPUT_DESC][0][FORMAT],
+                                                      op[OUTPUT_DESC][0][ORI_FORMAT])
+    elif op_name == "MatMul":
+        op_attrs[DST_TYPE] = op[OUTPUT_DESC][0][DATA_TYPE]
+        op_attrs["dst_format"] = op[OUTPUT_DESC][0][FORMAT]
+    elif op_name == "Cast":
+        op_attrs[DST_TYPE] = op[OUTPUT_DESC][0][DATA_TYPE]
+    op_attrs["fusion_op_name"] = fusion_op_name
+    return op_attrs
+
+
+def create_placeholders(inputs):
+    """Create placeholders."""
+    tensors = {}
+    for k, v in inputs.items():
+        dtype = v[DATA_TYPE]
+        if dtype == "bool":
+            dtype = "int8"
+        shape = v[SHAPE]
+        if NEW_SHAPE in v:
+            shape = v[NEW_SHAPE]
+        attr = {
+            FORMAT: v.get(FORMAT),
+            "sub_format": v.get("sub_format", ""),
+            ORI_SHAPE: v.get(ORI_SHAPE),
+            ORI_FORMAT: v.get(ORI_FORMAT),
+            "addr_type": v.get("addr_type", 0),
+            "valid_shape": v.get("valid_shape", []),
+            "slice_offset": v.get("slice_offset", []),
+            "L1_fusion_type": v.get("L1_fusion_type", -1),
+            "L1_addr_flag": v.get("L1_addr_flag", -1),
+            "L1_addr_offset": v.get("L1_addr_offset", -1),
+            "L1_valid_size": v.get("L1_valid_size", -1),
+            "range": v.get("range", [])
+        }
+        if attr.get(ORI_SHAPE) is None:
+            attr[ORI_SHAPE] = infer_ori_shape(v.get(SHAPE), v.get(FORMAT), attr.get(ORI_FORMAT))
+        tensors[k] = tvm.placeholder(shape=shape, name=k, dtype=dtype, attrs=attr)
+    return tensors
+
+
+def same_shape(inputs):
+    """Check if all inputs have same shape."""
+    if not inputs:
+        return True
+    base_shape = -1
+    for _, v in inputs.items():
+        if base_shape == -1:
+            base_shape = v[SHAPE]
+        if v[SHAPE] != base_shape:
+            return False
+    return True
+
+
+def create_input_tensors(json_dict):
+    """Create input placeholders."""
+    fold_dim = True
+    inputs = get_input_desc(json_dict.get(INPUT_DESC, []))
+    for op in json_dict["op_desc"]:
+        op_name = op["name"]
+        pattern = get_op_reg_info(op_name, "pattern")
+        op_inputs = get_input_desc(op.get(INPUT_DESC, []))
+        TransShape.run(op_name, pattern, op_inputs, inputs)
+        if pattern != OpPattern.ELEMWISE or not same_shape(op_inputs):
+            fold_dim = False
+    if fold_dim:
+        for k, v in inputs.items():
+            shape = v[SHAPE]
+            new_shape = [functools.reduce(lambda x, y: x * y, shape[:])]
+            add_new_shape(k, shape, new_shape, inputs)
+    return create_placeholders(inputs)
+
+
+def create_fusion_op_name(op_names):
+    """Create fusion op name, which is used in same op compute process."""
+    fusion_op_name = "te_fusion" if len(op_names) > 1 else ""
+    for op_name in op_names:
+        kernel_name = get_op_reg_info(op_name, "kernel_name")
+        fusion_op_name = fusion_op_name + "_" + kernel_name
+    return fusion_op_name
+
+
+def update_format(json_dict):
+    """Some format like DefaultFormat is not recognized in TBE, need to convert these formats."""
+
+    def _update_input_format(input_desc):
+        for desc in input_desc:
+            for item in desc:
+                if item[FORMAT] == "DefaultFormat":
+                    item[FORMAT] = "ND"
+                if item.get(ORI_FORMAT) is None or item[ORI_FORMAT] == "DefaultFormat":
+                    item[ORI_FORMAT] = "NCHW"
+
+    def _update_output_format(output_desc):
+        for item in output_desc:
+            if item[FORMAT] == "DefaultFormat":
+                item[FORMAT] = "ND"
+            if item.get(ORI_FORMAT) is None or item[ORI_FORMAT] == "DefaultFormat":
+                item[ORI_FORMAT] = "NCHW"
+
+    _update_input_format(json_dict.get(INPUT_DESC, []))
+    _update_output_format(json_dict[OUTPUT_DESC])
+    for op in json_dict["op_desc"]:
+        _update_input_format(op.get(INPUT_DESC, []))
+        _update_output_format(op[OUTPUT_DESC])
+
+
+def gen_args_remap(orig_inputs_name, orig_outputs_name, inputs_name, outputs_name, inplace_names):
+    """Generate the final kernel args indices."""
+    input_indices = []
+    output_indices = []
+    ni = len(orig_inputs_name)
+    for _, name in enumerate(inputs_name):
+        if name not in orig_inputs_name:
+            raise ValueError("Current input name [{}] can not be found in original input names list: {}"
+                             .format(name, orig_inputs_name))
+        input_indices.append(orig_inputs_name.index(name))
+    for _, name in enumerate(outputs_name):
+        if name in orig_outputs_name:
+            output_indices.append(ni + orig_outputs_name.index(name))
+        elif name in inplace_names and inplace_names[name] in orig_inputs_name:
+            output_indices.append(orig_inputs_name.index(inplace_names[name]))
+        else:
+            raise ValueError("Current output name [{}] can not be found in original output names list: {}, and it not "
+                             "inplace to an input tensor".format(name, orig_outputs_name))
+    return input_indices, output_indices
+
+
+def update_json(json_dict, inputs_name, outputs_name, inplace_names, kernel_meta_parent_dir):
+    """update kernel json."""
+
+    def _get_tensor_name(idx):
+        if idx < len(input_indices):
+            orig_idx = input_indices[idx]
+        elif idx - len(input_indices) < len(output_indices):
+            orig_idx = output_indices[idx - len(input_indices)]
+        else:
+            raise ValueError("parameters index >= the sum of input and output tensor numbers: {} vs {}"
+                             .format(idx, len(input_indices) + len(output_indices)))
+        if orig_idx < len(orig_input_names):
+            return orig_input_names[orig_idx]
+        if orig_idx - len(orig_input_names) >= len(orig_output_names):
+            raise ValueError("parameters index [{}]'s original index >= the sum of original input and original output"
+                             "tensor numbers: {} vs {}"
+                             .format(idx, orig_idx, len(orig_input_names) + len(orig_output_names)))
+        return orig_output_names[orig_idx - len(orig_input_names)]
+
+    def _update_parameters(cur_parameters):
+        atomic_tensors = set()
+        for i, p in enumerate(cur_parameters):
+            if p == 0:
+                continue
+            orig_tensor_name = _get_tensor_name(i)
+            if orig_tensor_name not in orig_input_names:
+                atomic_tensors.add(orig_tensor_name)
+        res = [0] * len(orig_input_names)
+        for _, name in enumerate(orig_output_names):
+            p = 1 if name in atomic_tensors else 0
+            res.append(p)
+        return res
+
+    def _save_json():
+        # remove original .json
+        try:
+            os.remove(json_path)
+        except OSError:
+            pass
+        # generate new .json
+        try:
+            with os.fdopen(os.open(json_path, os.O_WRONLY | os.O_CREAT | os.O_TRUNC, 0o600), 'w') as fi:
+                json.dump(json_dict, fi, sort_keys=True, indent=4, separators=(',', ':'))
+        except OSError:
+            pass
+
+    global_attrs = get_attr_dict(json_dict.get("global_attrs", []))
+    orig_input_names = global_attrs.get("orig_input_names", [])
+    orig_output_names = global_attrs.get("orig_output_names", [])
+    input_indices, output_indices = gen_args_remap(orig_input_names, orig_output_names,
+                                                   inputs_name, outputs_name, inplace_names)
+    json_name = json_dict.get("op") + ".json"
+    json_path = os.path.join(os.path.realpath(kernel_meta_parent_dir), "kernel_meta", json_name)
+    if os.path.isfile(json_path):
+        with open(json_path, "r") as f:
+            json_dict = json.loads(f.read())
+        # Save args_remap to .json: use this information to fetch the correct address during kernel launch
+        json_dict["args_remap"] = [input_indices, output_indices]
+        # Update parameters: current parameters is relative to current info(the composite optimized info) whose input
+        # and output tensors may be different from the original info, however, the frame can only see the original
+        # info's input and output tensors, so we need to generate a new parameters which is relative to original info.
+        parameters = json_dict.get("parameters", [])
+        new_parameters = _update_parameters(parameters)
+        json_dict["parameters"] = new_parameters
+        # Save the updated json
+        _save_json()
+
+
+def build(json_str, kernel_meta_parent_dir):
+    """Build kernel."""
+    json_dict = json.loads(json_str)
+    update_format(json_dict)
+    inputs_name = get_inputs_name(json_dict.get(INPUT_DESC, []))
+    outputs_name, inplace_names = get_outputs_info(json_dict[OUTPUT_DESC])
+    op_names = get_all_op_name(json_dict["op_desc"])
+    fusion_op_name = create_fusion_op_name(op_names)
+
+    # Create input placeholder
+    all_tensors = create_input_tensors(json_dict)
+
+    with op_context.OpContext("pre_static"):
+        context = op_context.get_context()
+        context.add_addition("op_bank_path", "")
+        context.add_addition("mdl_bank_path", "")
+
+        # Emit op
+        for op in json_dict["op_desc"]:
+            op_name = op["name"]
+            # get op input tensor
+            op_inputs = get_inputs_tensor(op.get(INPUT_DESC, []), all_tensors)
+            # get op attrs
+            op_attrs = get_op_attrs(op, fusion_op_name)
+            # op compute
+            op_outputs = get_op_reg_info(op_name, "func")(*op_inputs, attrs=op_attrs)
+            # update op output tensor
+            if not isinstance(op_outputs, (list, tuple)):
+                op_outputs = [op_outputs]
+            if len(op[OUTPUT_DESC]) != len(op_outputs):
+                raise ValueError("len(op[\"output_desc\"] is not equal to the number of real output tensors in op[{}]: "
+                                 "{} vs {}".format(op_name, len(op[OUTPUT_DESC]), len(op_outputs)))
+            for i, desc in enumerate(op[OUTPUT_DESC]):
+                all_tensors[desc["tensor_name"]] = op_outputs[i]
+
+        # Collect input, output tensors
+        io_tensors = []
+        output_tensors = []
+        for name in inputs_name:
+            io_tensors.append(all_tensors.get(name))
+        for name in outputs_name:
+            output_tensors.append(all_tensors.get(name))
+            io_tensors.append(all_tensors.get(name))
+
+        # Schedule and build
+        with tvm.target.cce():
+            sch = auto_schedule(output_tensors)
+        config = {"name": json_dict["op"],
+                  "tensor_list": io_tensors}
+        update_config(config, op_names)
+        tbe_build(sch, config)
+
+    if inplace_names:
+        update_json(json_dict, inputs_name, outputs_name, inplace_names, kernel_meta_parent_dir)
+
+
+def build_tbe_kernel(json_str, kernel_meta_parent_dir):
+    """Build TBE kernel."""
+    arch_name = get_arch_name(json.loads(json_str))
+    initialize(kernel_meta_parent_dir, arch_name)
+    with build_config(kernel_meta_parent_dir=kernel_meta_parent_dir):
+        build(json_str, kernel_meta_parent_dir)

mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/compiler.py

@@ -0,0 +1,56 @@
+# Copyright 2020-2021 Huawei Technologies Co., Ltd
+#
+# Licensed 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.
+# ============================================================================
+"""Providing akg compile with json"""
+import os
+import sys
+
+
+def run_compiler(op_json, compile_backend="AKG", attrs=None, kernel_meta_parent_dir="./"):
+    """
+    Compile `op_json` with the selected `compile_backend`.
+
+    Args:
+        op_json (str): json string of the op.
+        compile_backend (str): compilation backend, can be "AKG" or "TBE".
+        attrs (Union[str, dict]): compilation attributes. Used in "AKG" compile_backend.
+        kernel_meta_parent_dir (str): kernel_meta parent dir.
+
+    Returns:
+        None
+    """
+    if os.path.isfile(op_json):
+        with open(op_json, 'r') as f:
+            op_json = f.read()
+    # Compile op_json
+    if compile_backend == "TBE":
+        from build_tbe_kernel import build_tbe_kernel
+        build_tbe_kernel(op_json, kernel_meta_parent_dir)
+    else:
+        os.environ["MS_COMPILER_CACHE_PATH"] = kernel_meta_parent_dir
+        os.environ["KERNEL_META_DIR"] = "kernel_meta"
+        from get_file_path import get_akg_path
+        sys.path.insert(0, get_akg_path())
+        p = __import__("akg", globals(), locals(), ['ms'], 0)
+        func = getattr(p.ms, "compilewithjson")
+        res = func(op_json, attrs)
+        if not res:
+            raise ValueError("Compile error")
+
+
+if __name__ == "__main__":
+    if len(sys.argv) > 2:
+        run_compiler(sys.argv[1], *sys.argv[2:])
+    else:
+        run_compiler(sys.argv[1])