from cimodel.lib.conf_tree import ConfigNode, XImportant
from cimodel.lib.conf_tree import Ver
CONFIG_TREE_DATA = [
(Ver("ubuntu", "16.04"), [
([Ver("gcc", "5")], [XImportant("onnx_py2")]),
([Ver("clang", "7")], [
XImportant("onnx_main_py3.6"),
XImportant("onnx_ort1_py3.6"),
XImportant("onnx_ort2_py3.6")
]),
]),
]
class TreeConfigNode(ConfigNode):
def __init__(self, parent, node_name, subtree):
super(TreeConfigNode, self).__init__(parent,
self.modify_label(node_name))
self.subtree = subtree
self.init2(node_name)
# noinspection PyMethodMayBeStatic
def modify_label(self, label):
return str(label)
def init2(self, node_name):
pass
def get_children(self):
return [
from cimodel.lib.conf_tree import ConfigNode, X, XImportant
from cimodel.lib.conf_tree import Ver
CONFIG_TREE_DATA = [
(
Ver("ubuntu", "16.04"),
[
(
[Ver("cuda", "9.0")],
[
# TODO make explicit that this is a "secret TensorRT build"
# (see https://github.com/pytorch/pytorch/pull/17323#discussion_r259446749)
# TODO Uh oh, were we supposed to make this one important?!
X("py2"),
XImportant("cmake"),
]),
([Ver("cuda", "10.1")], [XImportant("py3.5")]), # TensorRT 6 build
([Ver("mkl")], [XImportant("py2")]),
([Ver("gcc", "5")], [XImportant("onnx_py2")]),
([Ver("clang", "3.8")], [X("py2")]),
([Ver("clang", "3.9")], [X("py2")]),
([Ver("clang",
"7")], [XImportant("py2"),
XImportant("onnx_py3.6")]),
([Ver("android")], [XImportant("py2")]),
]),
(Ver("centos", "7"), [
([Ver("devtoolset", "7"), Ver("cuda", "9.2")], [X("py3.6")]),
]),
(
Ver("macos", "10.13"),
from cimodel.lib.conf_tree import ConfigNode, XImportant
from cimodel.lib.conf_tree import Ver
CONFIG_TREE_DATA = [
(Ver("ubuntu", "16.04"), [
([Ver("clang", "7")], [
XImportant("onnx_main_py3.6"),
XImportant("onnx_ort1_py3.6"),
XImportant("onnx_ort2_py3.6")
]),
]),
]
class TreeConfigNode(ConfigNode):
def __init__(self, parent, node_name, subtree):
super(TreeConfigNode, self).__init__(parent,
self.modify_label(node_name))
self.subtree = subtree
self.init2(node_name)
# noinspection PyMethodMayBeStatic
def modify_label(self, label):
return str(label)
def init2(self, node_name):
pass
def get_children(self):
return [
self.child_constructor()(self, k, v) for (k, v) in self.subtree
#!/usr/bin/env python3
from collections import OrderedDict
import cimodel.data.dimensions as dimensions
import cimodel.lib.miniutils as miniutils
from cimodel.lib.conf_tree import Ver
DOCKER_IMAGE_PATH_BASE = "308535385114.dkr.ecr.us-east-1.amazonaws.com/caffe2/"
DOCKER_IMAGE_VERSION = 253
CONFIG_HIERARCHY = [
(Ver("ubuntu", "14.04"), [
(Ver("gcc", "4.8"), ["py2"]),
(Ver("gcc", "4.9"), ["py2"]),
]),
(
Ver("ubuntu", "16.04"),
[
(Ver("cuda", "8.0"), ["py2"]),
(
Ver("cuda", "9.0"),
[
# TODO make explicit that this is a "secret TensorRT build"
# (see https://github.com/pytorch/pytorch/pull/17323#discussion_r259446749)
"py2",
"cmake",
]),
(Ver("cuda", "9.1"), ["py2"]),
(Ver("mkl"), ["py2"]),
from cimodel.lib.conf_tree import ConfigNode, XImportant
from cimodel.lib.conf_tree import Ver
CONFIG_TREE_DATA = [
(
Ver("ubuntu", "16.04"),
[
# ([Ver("gcc", "5")], [X("onnx_py3.6")]),
([Ver("clang", "7")], [XImportant("onnx_py3.6")]),
]),
]
class TreeConfigNode(ConfigNode):
def __init__(self, parent, node_name, subtree):
super(TreeConfigNode, self).__init__(parent,
self.modify_label(node_name))
self.subtree = subtree
self.init2(node_name)
# noinspection PyMethodMayBeStatic
def modify_label(self, label):
return str(label)
def init2(self, node_name):
pass
def get_children(self):
return [
self.child_constructor()(self, k, v) for (k, v) in self.subtree
]