Python CudaVersion 예제들

예제 #1

파일: windows_build_definitions.py 프로젝트: wenhaopeter/read_pytorch_code

                          self.dotted_version()]))


def FalsePred(_):
    return False


def TruePred(_):
    return True


WORKFLOW_DATA = [
    # VS2017 CUDA-10.1
    WindowsJob(None,
               VcSpec(2017, ["14", "11"]),
               CudaVersion(10, 1),
               master_only_pred=FalsePred),
    WindowsJob(1, VcSpec(2017, ["14", "11"]), CudaVersion(10, 1)),
    # VS2017 no-CUDA (builds only)
    WindowsJob(None, VcSpec(2017, ["14", "16"]), CudaVersion(10, 1)),
    WindowsJob(None, VcSpec(2017, ["14", "16"]), None),
    # VS2019 CUDA-10.1
    WindowsJob(None, VcSpec(2019), CudaVersion(10, 1)),
    WindowsJob(1, VcSpec(2019), CudaVersion(10, 1)),
    WindowsJob(2, VcSpec(2019), CudaVersion(10, 1)),
    # VS2019 CPU-only
    WindowsJob(None, VcSpec(2019), None),
    WindowsJob(1, VcSpec(2019), None),
    WindowsJob(2, VcSpec(2019), None, master_only_pred=TruePred),
    WindowsJob(1, VcSpec(2019), CudaVersion(10, 1), force_on_cpu=True),
    WindowsJob(2,

예제 #2

        return "_".join(self.get_elements())


def FalsePred(_):
    return False


def TruePred(_):
    return True


_VC2019 = VcSpec(2019)

WORKFLOW_DATA = [
    # VS2019 CUDA-10.1
    WindowsJob(None, _VC2019, CudaVersion(10, 1)),
    WindowsJob(1, _VC2019, CudaVersion(10, 1)),
    WindowsJob(2, _VC2019, CudaVersion(10, 1)),
    # VS2019 CUDA-11.1
    WindowsJob(None, _VC2019, CudaVersion(11, 1)),
    WindowsJob(1, _VC2019, CudaVersion(11, 1), master_only_pred=TruePred),
    WindowsJob(2, _VC2019, CudaVersion(11, 1), master_only_pred=TruePred),
    # VS2019 CPU-only
    WindowsJob(None, _VC2019, None),
    WindowsJob(1, _VC2019, None, master_only_pred=TruePred),
    WindowsJob(2, _VC2019, None, master_only_pred=TruePred),
    WindowsJob(1,
               _VC2019,
               CudaVersion(10, 1),
               force_on_cpu=True,
               master_only_pred=TruePred),

예제 #3

파일: windows_build_definitions.py 프로젝트: ydcjeff/pytorch

            return [self.prefixed_year()]
        return [self.prefixed_year()] + self.version_elements

    def get_product(self):
        return "BuildTools"

    def dotted_version(self):
        return ".".join(self.version_elements)

    def prefixed_year(self):
        return "vs" + str(self.year)

    def render(self):
        return "_".join(self.get_elements())

_VC2019 = VcSpec(2019)

WORKFLOW_DATA = [
    # VS2019 CUDA-10.1
    WindowsJob(None, _VC2019, CudaVersion(10, 1), master_only=True),
    # VS2019 CUDA-10.1 force on cpu
    WindowsJob(1, _VC2019, CudaVersion(10, 1), force_on_cpu=True, master_only=True),

    # TODO: This test is disabled due to https://github.com/pytorch/pytorch/issues/59724
    # WindowsJob('_azure_multi_gpu', _VC2019, CudaVersion(11, 1), multi_gpu=True, master_and_nightly=True),
]


def get_windows_workflows():
    return [item.gen_tree() for item in WORKFLOW_DATA]

예제 #4

파일: ge_config_tests.py 프로젝트: zmmyc/pytorch

                 ["pytorch_linux_xenial_py3_6_gcc5_4_build"]),
 GeConfigTestJob(MultiPartVersion([3, 6], "py"),
                 MultiPartVersion([5, 4], "gcc"), None,
                 ["ge_config_profiling", "test"],
                 ["pytorch_linux_xenial_py3_6_gcc5_4_build"]),
 GeConfigTestJob(
     MultiPartVersion([3, 6], "py"),
     MultiPartVersion([5, 4], "gcc"),
     None,
     ["ge_config_simple", "test"],
     ["pytorch_linux_xenial_py3_6_gcc5_4_build"],
 ),
 GeConfigTestJob(
     None,
     None,
     CudaVersion(10, 2),
     ["cudnn7", "py3", "ge_config_legacy", "test"],
     ["pytorch_linux_xenial_cuda10_2_cudnn7_py3_gcc7_build"],
     use_cuda_docker=True,
     # TODO Why does the build environment specify cuda10.1, while the
     # job name is cuda10_2?
     build_env_override=
     "pytorch-linux-xenial-cuda10.1-cudnn7-ge_config_legacy-test"),
 GeConfigTestJob(
     None,
     None,
     CudaVersion(10, 2),
     ["cudnn7", "py3", "ge_config_profiling", "test"],
     ["pytorch_linux_xenial_cuda10_2_cudnn7_py3_gcc7_build"],
     use_cuda_docker=True,
     # TODO Why does the build environment specify cuda10.1, while the

예제 #5

    def dotted_version(self):
        return ".".join(self.version_elements)

    def prefixed_year(self):
        return "vs" + str(self.year)

    def render(self):
        return "_".join(self.get_elements())


_VC2019 = VcSpec(2019)

WORKFLOW_DATA = [
    # VS2019 CUDA-10.1
    WindowsJob(None, _VC2019, CudaVersion(10, 1), master_only=True),
    # VS2019 CUDA-10.1 force on cpu
    WindowsJob(1,
               _VC2019,
               CudaVersion(10, 1),
               force_on_cpu=True,
               master_only=True),

    # TODO: This test is disabled due to https://github.com/pytorch/pytorch/issues/59724
    # WindowsJob('_azure_multi_gpu', _VC2019, CudaVersion(11, 1), multi_gpu=True, master_and_nightly=True),
]


def get_windows_workflows():
    return [item.gen_tree() for item in WORKFLOW_DATA]

예제 #6

        return "_".join(
            filter(None, [self.prefixed_year(),
                          self.dotted_version()]))


def FalsePred(_):
    return False


def TruePred(_):
    return True


WORKFLOW_DATA = [
    # VS2019 CUDA-10.1
    WindowsJob(None, VcSpec(2019), CudaVersion(10, 1)),
    WindowsJob(1, VcSpec(2019), CudaVersion(10, 1)),
    WindowsJob(2, VcSpec(2019), CudaVersion(10, 1)),
    # VS2019 CUDA-11.0
    WindowsJob(None, VcSpec(2019), CudaVersion(11, 0)),
    WindowsJob(1, VcSpec(2019), CudaVersion(11, 0)),
    WindowsJob(2, VcSpec(2019), CudaVersion(11, 0)),
    # VS2019 CPU-only
    WindowsJob(None, VcSpec(2019), None),
    WindowsJob(1, VcSpec(2019), None, master_only_pred=TruePred),
    WindowsJob(2, VcSpec(2019), None, master_only_pred=TruePred),
    WindowsJob(1,
               VcSpec(2019),
               CudaVersion(10, 1),
               force_on_cpu=True,
               master_only_pred=TruePred),

예제 #7

def FalsePred(_):
    return False


def TruePred(_):
    return True


# MKLDNN compilation fails with VC-19.27
_VC2019 = VcSpec(2019, ["14", "26"], hide_version=True)

WORKFLOW_DATA = [
    # VS2019 CUDA-10.1
    WindowsJob(None, _VC2019, CudaVersion(10, 1)),
    WindowsJob(1, _VC2019, CudaVersion(10, 1)),
    WindowsJob(2, _VC2019, CudaVersion(10, 1)),
    # VS2019 CUDA-11.0
    WindowsJob(None, _VC2019, CudaVersion(11, 0)),
    WindowsJob(1, _VC2019, CudaVersion(11, 0)),
    WindowsJob(2, _VC2019, CudaVersion(11, 0)),
    # VS2019 CPU-only
    WindowsJob(None, _VC2019, None),
    WindowsJob(1, _VC2019, None, master_only_pred=TruePred),
    WindowsJob(2, _VC2019, None, master_only_pred=TruePred),
    WindowsJob(1,
               _VC2019,
               CudaVersion(10, 1),
               force_on_cpu=True,
               master_only_pred=TruePred),

예제 #8

                          self.dotted_version()]))


def FalsePred(_):
    return False


def TruePred(_):
    return True


WORKFLOW_DATA = [
    # VS2017 CUDA-10.1
    WindowsJob(None,
               VcSpec(2017, ["14", "13"]),
               CudaVersion(10, 1),
               master_only_pred=FalsePred),
    WindowsJob(1, VcSpec(2017, ["14", "13"]), CudaVersion(10, 1)),
    # VS2017 no-CUDA (builds only)
    WindowsJob(None, VcSpec(2017, ["14", "16"]), None),
    # VS2019 CUDA-10.1
    WindowsJob(None, VcSpec(2019), CudaVersion(10, 1)),
    WindowsJob(1, VcSpec(2019), CudaVersion(10, 1)),
    WindowsJob(2, VcSpec(2019), CudaVersion(10, 1)),
    # VS2019 CPU-only
    WindowsJob(None, VcSpec(2019), None),
    WindowsJob(1, VcSpec(2019), None, master_only_pred=TruePred),
    WindowsJob(2, VcSpec(2019), None, master_only_pred=TruePred),
    WindowsJob(1,
               VcSpec(2019),
               CudaVersion(10, 1),

예제 #9

    def prefixed_year(self):
        return "vs" + str(self.year)

    def render(self):
        return "_".join(filter(None, [self.prefixed_year(), self.dotted_version()]))

def FalsePred(_):
    return False

def TruePred(_):
    return True

WORKFLOW_DATA = [
    # VS2019 CUDA-10.1
    WindowsJob(None, VcSpec(2019), CudaVersion(10, 1)),
    WindowsJob(1, VcSpec(2019), CudaVersion(10, 1)),
    WindowsJob(2, VcSpec(2019), CudaVersion(10, 1)),
    WindowsJob("-jit-profiling-tests", VcSpec(2019), CudaVersion(10, 1), master_only_pred=FalsePred),
    # VS2019 CUDA-11.0
    WindowsJob(None, VcSpec(2019), CudaVersion(11, 0)),
    WindowsJob(1, VcSpec(2019), CudaVersion(11, 0)),
    WindowsJob(2, VcSpec(2019), CudaVersion(11, 0)),
    WindowsJob("-jit-profiling-tests", VcSpec(2019), CudaVersion(11, 0), master_only_pred=FalsePred),
    # VS2019 CPU-only
    WindowsJob(None, VcSpec(2019), None),
    WindowsJob(1, VcSpec(2019), None, master_only_pred=TruePred),
    WindowsJob(2, VcSpec(2019), None, master_only_pred=TruePred),
    WindowsJob(1, VcSpec(2019), CudaVersion(10, 1), force_on_cpu=True, master_only_pred=TruePred),
]

예제 #10

        return "vs" + str(self.year)

    def render(self):
        return "_".join(self.get_elements())

def FalsePred(_):
    return False

def TruePred(_):
    return True

_VC2019 = VcSpec(2019)

WORKFLOW_DATA = [
    # VS2019 CUDA-10.1
    WindowsJob(None, _VC2019, CudaVersion(10, 1)),
    # Disable windows tests until https://github.com/pytorch/pytorch/issues/56654 is resolved
    # WindowsJob(1, _VC2019, CudaVersion(10, 1)),
    # WindowsJob(2, _VC2019, CudaVersion(10, 1)),
    # VS2019 CUDA-11.1
    WindowsJob(None, _VC2019, CudaVersion(11, 1)),
    # Disable windows tests until https://github.com/pytorch/pytorch/issues/56654 is resolved
    # WindowsJob(1, _VC2019, CudaVersion(11, 1), master_only_pred=TruePred),
    # WindowsJob(2, _VC2019, CudaVersion(11, 1), master_only_pred=TruePred),
    # VS2019 CPU-only
    WindowsJob(None, _VC2019, None),
    # Remove master only predicate until GPU tests can be re-enabled again
    WindowsJob(1, _VC2019, None),
    WindowsJob(2, _VC2019, None),
    WindowsJob(1, _VC2019, CudaVersion(10, 1), force_on_cpu=True, master_only_pred=TruePred),
]