예제 #1
0
from code import main
#from code import main

main.main()
예제 #2
0
    for system in systems:
        print("Generating measurements/ entries for", system)

        config_ver_keywords = ["maxq", "high_accuracy"]
        if "Xavier" not in system.get_id():
            config_ver_keywords += ["triton"]

        cross_products = itertools.product([False, True],
                                           repeat=len(config_ver_keywords))
        all_config_vers = []
        for config_product in cross_products:
            if not any(config_product):
                all_config_vers.append("default")
            else:
                all_config_vers.append("_".join(
                    itertools.compress(config_ver_keywords, config_product)))

        if "MIG_1x1g" in system.get_id():
            all_config_vers.extend(["hetero", "hetero_high_accuracy"])

        # 56 MIG slice only uses Triton, not default config_ver
        if "MIG_56x1g" in system.get_id():
            all_config_vers.remove("default")

        # Triton CPU has special config_vers, and does not use 'default'
        if system.get_id().startswith("Triton_CPU"):
            all_config_vers = ["openvino", "openvino_high_accuracy"]

        default_args["config_ver"] = ",".join(all_config_vers)
        main(default_args, system)
# 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.

import sys, os, re
sys.path.insert(0, os.getcwd())

from code.common.system_list import system_list
from code.common import dict_get
from code.main import main

if __name__ == "__main__":
    default_args = {
        "action": "generate_conf_files",
        "benchmarks": None,  # Defaults to all benchmarks
        "scenarios": None,  # Defaults to all scenarios
        "config_ver": "all",
        "configs": "",
        "no_gpu": False,
        "gpu_only": False
    }

    systems = set([x[0] for x in system_list])
    for system_id in systems:
        print("Generating measurements/ entries for", system_id)
        if "Xavier" in system_id:
            default_args["config_ver"] = "default,high_accuracy"
        else:
            default_args["config_ver"] = "all"
        main(default_args, system_id)