+++ /dev/null
-import cimodel.lib.miniutils as miniutils
-from cimodel.data.simple.util.versions import MultiPartVersion, CudaVersion
-from cimodel.data.simple.util.docker_constants import DOCKER_IMAGE_BASIC, DOCKER_IMAGE_CUDA_10_2
-
-
-class GeConfigTestJob:
- def __init__(self,
- py_version,
- gcc_version,
- cuda_version,
- variant_parts,
- extra_requires,
- use_cuda_docker=False,
- build_env_override=None):
-
- self.py_version = py_version
- self.gcc_version = gcc_version
- self.cuda_version = cuda_version
- self.variant_parts = variant_parts
- self.extra_requires = extra_requires
- self.use_cuda_docker = use_cuda_docker
- self.build_env_override = build_env_override
-
- def get_all_parts(self, with_dots):
-
- maybe_py_version = self.py_version.render_dots_or_parts(with_dots) if self.py_version else []
- maybe_gcc_version = self.gcc_version.render_dots_or_parts(with_dots) if self.gcc_version else []
- maybe_cuda_version = self.cuda_version.render_dots_or_parts(with_dots) if self.cuda_version else []
-
- common_parts = [
- "pytorch",
- "linux",
- "xenial",
- ] + maybe_cuda_version + maybe_py_version + maybe_gcc_version
-
- return common_parts + self.variant_parts
-
- def gen_tree(self):
-
- resource_class = "gpu.medium" if self.use_cuda_docker else "large"
- docker_image = DOCKER_IMAGE_CUDA_10_2 if self.use_cuda_docker else DOCKER_IMAGE_BASIC
- full_name = "_".join(self.get_all_parts(False))
- build_env = self.build_env_override or "-".join(self.get_all_parts(True))
-
- props_dict = {
- "name": full_name,
- "build_environment": build_env,
- "requires": self.extra_requires,
- "resource_class": resource_class,
- "docker_image": docker_image,
- }
-
- if self.use_cuda_docker:
- props_dict["use_cuda_docker_runtime"] = miniutils.quote(str(1))
-
- return [{"pytorch_linux_test": props_dict}]
-
-
-WORKFLOW_DATA = [
- GeConfigTestJob(
- MultiPartVersion([3, 6], "py"),
- MultiPartVersion([5, 4], "gcc"),
- None,
- ["jit_legacy", "test"],
- ["pytorch_linux_xenial_py3_6_gcc5_4_build"]),
- GeConfigTestJob(
- None,
- None,
- CudaVersion(10, 2),
- ["cudnn7", "py3", "jit_legacy", "test"],
- ["pytorch_linux_xenial_cuda10_2_cudnn7_py3_gcc7_build"],
- use_cuda_docker=True,
- ),
-]
-
-
-def get_workflow_jobs():
- return [item.gen_tree() for item in WORKFLOW_DATA]
name: pytorch_linux_xenial_py3_clang5_mobile_code_analysis
requires:
- docker-pytorch-linux-xenial-py3-clang5-android-ndk-r19c
- - pytorch_linux_test:
- build_environment: pytorch-linux-xenial-py3.6-gcc5.4-jit_legacy-test
- docker_image: 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-py3.6-gcc5.4
- name: pytorch_linux_xenial_py3_6_gcc5_4_jit_legacy_test
- requires:
- - pytorch_linux_xenial_py3_6_gcc5_4_build
- resource_class: large
- - pytorch_linux_test:
- build_environment: pytorch-linux-xenial-cuda10.2-cudnn7-py3-jit_legacy-test
- docker_image: 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda10.2-cudnn7-py3-gcc7
- name: pytorch_linux_xenial_cuda10_2_cudnn7_py3_jit_legacy_test
- requires:
- - pytorch_linux_xenial_cuda10_2_cudnn7_py3_gcc7_build
- resource_class: gpu.medium
- use_cuda_docker_runtime: "1"
- binary_linux_build:
build_environment: manywheel 3.7m cu102 devtoolset7
docker_image: pytorch/manylinux-cuda102
import cimodel.data.simple.android_definitions
import cimodel.data.simple.binary_smoketest
import cimodel.data.simple.docker_definitions
-import cimodel.data.simple.ge_config_tests
import cimodel.data.simple.ios_definitions
import cimodel.data.simple.macos_definitions
import cimodel.data.simple.mobile_definitions
cimodel.data.simple.android_definitions.get_workflow_jobs,
cimodel.data.simple.ios_definitions.get_workflow_jobs,
cimodel.data.simple.mobile_definitions.get_workflow_jobs,
- cimodel.data.simple.ge_config_tests.get_workflow_jobs,
cimodel.data.simple.binary_smoketest.get_workflow_jobs,
cimodel.data.simple.nightly_ios.get_workflow_jobs,
cimodel.data.simple.nightly_android.get_workflow_jobs,