from cimodel.lib.conf_tree import ConfigNode, X, XImportant
CONFIG_TREE_DATA = [
("xenial", [
(None, [
X("nightly"),
]),
("gcc", [
("5.4", [ # All this subtree rebases to master and then build
XImportant("3.6"),
("3.6", [
("parallel_tbb", [X(True)]),
("parallel_native", [X(True)]),
]),
]),
# TODO: bring back libtorch test
("7", [X("3.6")]),
]),
("clang", [
("5", [
XImportant("3.6"), # This is actually the ASAN build
]),
]),
("cuda", [
("9", [
# Note there are magic strings here
# https://github.com/pytorch/pytorch/blob/master/.jenkins/pytorch/build.sh#L21
# and
# https://github.com/pytorch/pytorch/blob/master/.jenkins/pytorch/build.sh#L143
# and
from cimodel.lib.conf_tree import ConfigNode, X, XImportant
CONFIG_TREE_DATA = [
(
"xenial",
[
(
"gcc",
[
(
"5.4",
[ # All this subtree rebases to master and then build
("3.6", [
("important", [X(True)]),
("parallel_tbb", [X(True)]),
("parallel_native", [X(True)]),
("pure_torch", [X(True)]),
]),
]),
# TODO: bring back libtorch test
("7", [X("3.6")]),
]),
("clang", [
("5", [
("3.6", [
("asan", [
(True, [
("shard_test", [XImportant(True)]),
]),
]),
]),
from cimodel.lib.conf_tree import ConfigNode, X, XImportant
CONFIG_TREE_DATA = [
(
"xenial",
[
(
"gcc",
[
(
"5.4",
[ # All this subtree rebases to master and then build
("3.6", [
("important", [X(True)]),
("parallel_tbb", [X(True)]),
]),
]),
# TODO: bring back libtorch test
("7", [X("3.6")]),
]),
("clang", [
("7", [
("3.6", [
("asan", [
(True, [
("shard_test", [XImportant(True)]),
]),
]),
("onnx", [XImportant(True)]),
]),
]),
from cimodel.lib.conf_tree import ConfigNode, X, XImportant
CONFIG_TREE_DATA = [
(
"xenial",
[
(
None,
[
XImportant("2.7.9"),
X("2.7"),
XImportant(
"3.5"
), # Not run on all PRs, but should be included on [test all]
X("nightly"),
]),
(
"gcc",
[
(
"5.4",
[ # All this subtree rebases to master and then build
XImportant("3.6"),
("3.6", [
("parallel_tbb", [XImportant(True)]),
("parallel_native", [XImportant(True)]),
]),
]),
# TODO: bring back libtorch test
("7", [X("3.6")]),
]),
from cimodel.lib.conf_tree import ConfigNode, X, XImportant
CONFIG_TREE_DATA = [
("xenial", [
(None, [
X("nightly"),
]),
("gcc", [
("5.4", [ # All this subtree rebases to master and then build
XImportant("3.6"),
("3.6", [
("parallel_tbb", [X(True)]),
("parallel_native", [X(True)]),
]),
]),
# TODO: bring back libtorch test
("7", [X("3.6")]),
]),
("clang", [
("5", [
XImportant("3.6"), # This is actually the ASAN build
]),
]),
("cuda", [
("9.2", [
X("3.6"),
("3.6", [
("cuda_gcc_override", [X("gcc5.4")])
])
]),
from cimodel.lib.conf_tree import ConfigNode, X, XImportant
CONFIG_TREE_DATA = [
(
"xenial",
[
("rocm", [
("3.5.1", [
X("3.6"),
]),
]),
(
"gcc",
[
(
"5.4",
[ # All this subtree rebases to master and then build
("3.6", [
("important", [X(True)]),
("parallel_tbb", [X(True)]),
("parallel_native", [X(True)]),
]),
]),
# TODO: bring back libtorch test
("7", [X("3.6")]),
]),
("clang", [
("5", [
("3.6", [
("asan", [XImportant(True)]),
]),
#!/usr/bin/env python3
from cimodel.lib.conf_tree import ConfigNode, X
CONFIG_TREE_DATA = [
("trusty", [
(None, [
X("2.7.9"),
X("2.7"),
X("3.5"),
X("nightly"),
]),
("gcc", [
("4.8", [X("3.6")]),
("5.4", [
X("3.6"),
("3.6", [
("xla", [X(True)]),
("namedtensor", [X(True)]),
]),
]),
("7", [X("3.6")]),
]),
]),
("xenial", [
("clang", [
("5", [X("3.6")]),
]),
("cuda", [
("9", [
#!/usr/bin/env python3
from cimodel.lib.conf_tree import ConfigNode, X, XImportant
from cimodel.lib.conf_tree import Ver
CONFIG_TREE_DATA = [
(Ver("ubuntu", "14.04"), [
(Ver("gcc", "4.8"), [X("py2")]),
(Ver("gcc", "4.9"), [X("py2")]),
]),
(
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", "9.1"), [XImportant("py2")]),
(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")]),
]),
#!/usr/bin/env python3
from cimodel.lib.conf_tree import ConfigNode, X
from cimodel.lib.conf_tree import Ver
import cimodel.data.dimensions as dimensions
CONFIG_TREE_DATA = [
(Ver("ubuntu", "14.04"), [
(Ver("gcc", "4.8"), [X("py2")]),
(Ver("gcc", "4.9"), [X("py2")]),
]),
(
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)
X("py2"),
X("cmake"),
]),
(Ver("cuda", "9.1"), [X("py2")]),
(Ver("mkl"), [X("py2")]),
(Ver("gcc", "5"), [X("onnx_py2")]),
(Ver("clang", "3.8"), [X("py2")]),
(Ver("clang", "3.9"), [X("py2")]),
(Ver("clang", "7"), [X("py2")]),
(Ver("android"), [X("py2")]),
]),
(Ver("centos", "7"), [
#!/usr/bin/env python3
from cimodel.lib.conf_tree import ConfigNode, X, XImportant
CONFIG_TREE_DATA = [
("trusty", [
(None, [
XImportant("2.7.9"),
X("2.7"),
X("3.5"),
X("nightly"),
]),
("gcc", [
("4.8", [X("3.6")]),
("5.4", [
XImportant("3.6"),
("3.6", [
("xla", [XImportant(True)]),
("namedtensor", [XImportant(True)]),
]),
]),
("7", [X("3.6")]),
]),
]),
(
"xenial",
[
(
"clang",
[
(
from cimodel.lib.conf_tree import ConfigNode, X, XImportant
CONFIG_TREE_DATA = [
(
"xenial",
[
("rocm", [
("3.5.1", [
X("3.6"),
]),
]),
(
"gcc",
[
(
"5.4",
[ # All this subtree rebases to master and then build
XImportant("3.6"),
("3.6", [
("parallel_tbb", [X(True)]),
("parallel_native", [X(True)]),
]),
]),
# TODO: bring back libtorch test
("7", [X("3.6")]),
]),
(
"clang",
[
(
"5",
from cimodel.lib.conf_tree import ConfigNode, X, XImportant
CONFIG_TREE_DATA = [
(
"xenial",
[
(
"gcc",
[
(
"5.4",
[ # All this subtree rebases to master and then build
("3.6", [
("important", [X(True)]),
("parallel_tbb", [X(True)]),
("parallel_native", [X(True)]),
("pure_torch", [X(True)]),
]),
]),
# TODO: bring back libtorch test
("7", [X("3.6")]),
]),
("clang", [
("7", [
("3.6", [
("asan", [
(True, [
("shard_test", [XImportant(True)]),
]),
]),
("onnx", [XImportant(True)]),
from cimodel.lib.conf_tree import ConfigNode, X, XImportant
CONFIG_TREE_DATA = [
(
"xenial",
[
(None, [
X("nightly"),
]),
("rocm", [
("3.3", [
X("3.6"),
]),
]),
(
"gcc",
[
(
"5.4",
[ # All this subtree rebases to master and then build
XImportant("3.6"),
("3.6", [
("parallel_tbb", [X(True)]),
("parallel_native", [X(True)]),
]),
]),
# TODO: bring back libtorch test
("7", [X("3.6")]),
]),
(
"clang",
from cimodel.lib.conf_tree import ConfigNode, X
CONFIG_TREE_DATA = [
(
"xenial",
[
(
"gcc",
[
(
"5.4",
[ # All this subtree rebases to master and then build
("3.6", [
("important", [X(True)]),
]),
]),
# TODO: bring back libtorch test
]),
]),
]
def get_major_pyver(dotted_version):
parts = dotted_version.split(".")
return "py" + parts[0]
class TreeConfigNode(ConfigNode):
def __init__(self, parent, node_name, subtree):
super(TreeConfigNode, self).__init__(parent,
self.modify_label(node_name))
from cimodel.lib.conf_tree import ConfigNode, XImportant, X
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
]
def is_build_only(self):
from cimodel.lib.conf_tree import ConfigNode, X, XImportant
CONFIG_TREE_DATA = [
(
"xenial",
[
(
None,
[
XImportant("2.7.9"),
X("2.7"),
XImportant(
"3.5"
), # Not run on all PRs, but should be included on [test all]
X("nightly"),
]),
(
"gcc",
[
(
"5.4",
[ # All this subtree rebases to master and then build
XImportant("3.6"),
("3.6", [
("namedtensor", [XImportant(True)]),
]),
]),
# TODO: bring back libtorch test
("7", [X("3.6")]),
]),
(