Skip to content

Commit 4092d69

Browse files
mvsuspmvsusp
committed
Add PY2 support
1 parent ddf7475 commit 4092d69

File tree

3 files changed

+45
-22
lines changed

3 files changed

+45
-22
lines changed

‎buildspec.yml

Lines changed: 21 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ version: 0.2
33
env:
44
variables:
55
FRAMEWORK_VERSION: '1.1.0'
6-
CPU_PY_VERSION: '3'
6+
CPU_PY_VERSION: '2'
77
CPU_INSTANCE_TYPE: 'ml.c4.xlarge'
88
GPU_PY_VERSION: '3'
99
GPU_INSTANCE_TYPE: 'ml.p2.xlarge'
@@ -35,34 +35,44 @@ phases:
3535

3636
# build cpu base image
3737
- base_dir="docker/$FRAMEWORK_VERSION/base"
38+
- cpu_py2_base_tag="$FRAMEWORK_VERSION-cpu-py2"
3839
- cpu_py3_base_tag="$FRAMEWORK_VERSION-cpu-py3"
3940
- cpu_dockerfile="Dockerfile.cpu"
4041
- cd $base_dir
41-
- docker build -t $LOCAL_BASE_REPO:$cpu_py3_base_tag -f $cpu_dockerfile .
42+
- docker build -t $LOCAL_BASE_REPO:$cpu_py2_base_tag -f $cpu_dockerfile --build-arg py_version=2 .
43+
- docker build -t $LOCAL_BASE_REPO:$cpu_py3_base_tag -f $cpu_dockerfile --build-arg py_version=3 .
4244
- cd ../../../
4345

4446
# build gpu base image
47+
- gpu_py2_base_tag="$FRAMEWORK_VERSION-gpu-py2"
4548
- gpu_py3_base_tag="$FRAMEWORK_VERSION-gpu-py3"
4649
- gpu_dockerfile="Dockerfile.gpu"
4750
- cd $base_dir
48-
- docker build -t $LOCAL_BASE_REPO:$gpu_py3_base_tag -f $gpu_dockerfile .
51+
- docker build -t $LOCAL_BASE_REPO:$gpu_py2_base_tag -f $gpu_dockerfile --build-arg py_version=2 .
52+
- docker build -t $LOCAL_BASE_REPO:$gpu_py3_base_tag -f $gpu_dockerfile --build-arg py_version=3 .
4953
- cd ../../../
5054

5155
# create wheel
5256
- python3 setup.py bdist_wheel
5357

5458
# build cpu image
5559
- build_dir="docker/$FRAMEWORK_VERSION/final"
60+
- CPU_PY2_TAG="$FRAMEWORK_VERSION-cpu-py2-$BUILD_ID"
5661
- CPU_PY3_TAG="$FRAMEWORK_VERSION-cpu-py3-$BUILD_ID"
57-
- docker build -f "$build_dir/$cpu_dockerfile" -t $PREPROD_IMAGE:$CPU_PY3_TAG .
62+
- docker build -f "$build_dir/$cpu_dockerfile" --build-arg py_version=2 -t $PREPROD_IMAGE:$CPU_PY2_TAG .
63+
- docker build -f "$build_dir/$cpu_dockerfile" --build-arg py_version=3 -t $PREPROD_IMAGE:$CPU_PY3_TAG .
5864

5965
# build gpu image
66+
- GPU_PY2_TAG="$FRAMEWORK_VERSION-gpu-py2-$BUILD_ID"
6067
- GPU_PY3_TAG="$FRAMEWORK_VERSION-gpu-py3-$BUILD_ID"
61-
- docker build -f "$build_dir/$gpu_dockerfile" -t $PREPROD_IMAGE:$GPU_PY3_TAG .
68+
- docker build -f "$build_dir/$gpu_dockerfile" --build-arg py_version=2 -t $PREPROD_IMAGE:$GPU_PY2_TAG .
69+
- docker build -f "$build_dir/$gpu_dockerfile" --build-arg py_version=3 -t $PREPROD_IMAGE:$GPU_PY3_TAG .
6270

6371
# push images to ecr
6472
- $(aws ecr get-login --registry-ids $ACCOUNT --no-include-email --region $AWS_DEFAULT_REGION)
73+
- docker push $PREPROD_IMAGE:$CPU_PY2_TAG
6574
- docker push $PREPROD_IMAGE:$CPU_PY3_TAG
75+
- docker push $PREPROD_IMAGE:$GPU_PY2_TAG
6676
- docker push $PREPROD_IMAGE:$GPU_PY3_TAG
6777

6878
# launch remote gpu instance
@@ -74,7 +84,7 @@ phases:
7484
# run cpu integration tests
7585
- |
7686
if has-matching-changes "test/" "tests/" "src/*.py" "docker/*" "buildspec.yml"; then
77-
pytest test/integration/local --region $AWS_DEFAULT_REGION --docker-base-name $PREPROD_IMAGE --framework-version $FRAMEWORK_VERSION --processor cpu --tag $CPU_PY3_TAG
87+
pytest test/integration/local --region $AWS_DEFAULT_REGION --docker-base-name $PREPROD_IMAGE --framework-version $FRAMEWORK_VERSION --py-version $CPU_PY_VERSION --processor cpu
7888
else
7989
echo "skipping cpu integration tests"
8090
fi
@@ -83,7 +93,7 @@ phases:
8393
- |
8494
if has-matching-changes "test/" "tests/" "src/*.py" "docker/*" "buildspec.yml"; then
8595
printf "$SETUP_CMDS" > $SETUP_FILE
86-
py3_cmd="pytest test/integration/local --region $AWS_DEFAULT_REGION --docker-base-name $PREPROD_IMAGE --framework-version $FRAMEWORK_VERSION --processor gpu --tag $GPU_PY3_TAG"
96+
py3_cmd="pytest test/integration/local --region $AWS_DEFAULT_REGION --docker-base-name $PREPROD_IMAGE --framework-version $FRAMEWORK_VERSION --py-version $GPU_PY_VERSION --processor gpu"
8797
remote-test --github-repo $GITHUB_REPO --test-cmd "$py3_cmd" --setup-file $SETUP_FILE --pr-number "$PR_NUM"
8898
else
8999
echo "skipping gpu integration tests"
@@ -92,15 +102,15 @@ phases:
92102
# run cpu sagemaker tests
93103
- |
94104
if has-matching-changes "test/" "tests/" "src/*.py" "docker/*" "buildspec.yml"; then
95-
pytest test/integration/sagemaker --region $AWS_DEFAULT_REGION --docker-base-name $ECR_REPO --aws-id $ACCOUNT --framework-version $FRAMEWORK_VERSION --processor cpu --instance-type $CPU_INSTANCE_TYPE --tag $CPU_PY3_TAG
105+
pytest test/integration/sagemaker --region $AWS_DEFAULT_REGION --docker-base-name $ECR_REPO --aws-id $ACCOUNT --framework-version $FRAMEWORK_VERSION --py-version $CPU_PY_VERSION --processor cpu --instance-type $CPU_INSTANCE_TYPE
96106
else
97107
echo "skipping cpu sagemaker tests"
98108
fi
99109
100110
# run gpu sagemaker tests
101111
- |
102112
if has-matching-changes "test/" "tests/" "src/*.py" "docker/*" "buildspec.yml"; then
103-
pytest test/integration/sagemaker --region $AWS_DEFAULT_REGION --docker-base-name $ECR_REPO --aws-id $ACCOUNT --framework-version $FRAMEWORK_VERSION --processor gpu --instance-type $GPU_INSTANCE_TYPE --tag $GPU_PY3_TAG
113+
pytest test/integration/sagemaker --region $AWS_DEFAULT_REGION --docker-base-name $ECR_REPO --aws-id $ACCOUNT --framework-version $FRAMEWORK_VERSION --py-version $GPU_PY_VERSION --processor gpu --instance-type $GPU_INSTANCE_TYPE
104114
else
105115
echo "skipping gpu sagemaker tests"
106116
fi
@@ -111,5 +121,7 @@ phases:
111121
- cleanup-key-pairs
112122

113123
# remove ecr image
124+
- aws ecr batch-delete-image --repository-name $ECR_REPO --region $AWS_DEFAULT_REGION --image-ids imageTag=$CPU_PY2_TAG
114125
- aws ecr batch-delete-image --repository-name $ECR_REPO --region $AWS_DEFAULT_REGION --image-ids imageTag=$CPU_PY3_TAG
126+
- aws ecr batch-delete-image --repository-name $ECR_REPO --region $AWS_DEFAULT_REGION --image-ids imageTag=$GPU_PY2_TAG
115127
- aws ecr batch-delete-image --repository-name $ECR_REPO --region $AWS_DEFAULT_REGION --image-ids imageTag=$GPU_PY3_TAG

‎test/conftest.py

Lines changed: 17 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,10 @@
3737

3838
dir_path = os.path.dirname(os.path.realpath(__file__))
3939

40-
NO_P2_REGIONS = ['ap-northeast-3', 'ap-southeast-2', 'ca-central-1', 'eu-central-1',
41-
'eu-west-2', 'us-west-1']
42-
NO_P3_REGIONS = ['ap-northeast-3', 'ap-southeast-1', 'ap-southeast-2', 'ap-south-1', 'ca-central-1',
43-
'eu-central-1', 'eu-west-2', 'us-west-1']
40+
NO_P2_REGIONS = ['ap-east-1', 'ap-northeast-3', 'ap-southeast-2', 'ca-central-1', 'eu-central-1', 'eu-north-1',
41+
'eu-west-2', 'eu-west-3', 'us-west-1', 'sa-east-1']
42+
NO_P3_REGIONS = ['ap-east-1', 'ap-northeast-3', 'ap-southeast-1', 'ap-southeast-2', 'ap-south-1', 'ca-central-1',
43+
'eu-central-1', 'eu-north-1', 'eu-west-2', 'eu-west-3', 'sa-east-1', 'us-west-1']
4444

4545

4646
def pytest_addoption(parser):
@@ -51,8 +51,9 @@ def pytest_addoption(parser):
5151
parser.addoption('--docker-base-name', default='pytorch')
5252
parser.addoption('--region', default='us-west-2')
5353
parser.addoption('--framework-version', default=PyTorch.LATEST_VERSION)
54+
parser.addoption('--py-version', choices=['2', '3'], default=str(sys.version_info.major))
5455
parser.addoption('--processor', choices=['gpu', 'cpu'], default='cpu')
55-
# If not specified, will default to {framework-version}-{processor}-py3
56+
# If not specified, will default to {framework-version}-{processor}-py{py-version}
5657
parser.addoption('--tag', default=None)
5758

5859

@@ -71,15 +72,20 @@ def fixture_framework_version(request):
7172
return request.config.getoption('--framework-version')
7273

7374

75+
@pytest.fixture(scope='session', name='py_version')
76+
def fixture_py_version(request):
77+
return 'py{}'.format(int(request.config.getoption('--py-version')))
78+
79+
7480
@pytest.fixture(scope='session', name='processor')
7581
def fixture_processor(request):
7682
return request.config.getoption('--processor')
7783

7884

7985
@pytest.fixture(scope='session', name='tag')
80-
def fixture_tag(request, framework_version, processor):
86+
def fixture_tag(request, framework_version, processor, py_version):
8187
provided_tag = request.config.getoption('--tag')
82-
default_tag = '{}-{}-py3'.format(framework_version, processor)
88+
default_tag = '{}-{}-{}'.format(framework_version, processor, py_version)
8389
return provided_tag if provided_tag else default_tag
8490

8591

@@ -107,11 +113,12 @@ def fixture_use_gpu(processor):
107113

108114

109115
@pytest.fixture(scope='session', name='build_base_image', autouse=True)
110-
def fixture_build_base_image(request, framework_version, processor, tag, docker_base_name):
116+
def fixture_build_base_image(request, framework_version, py_version, processor, tag, docker_base_name):
111117
build_base_image = request.config.getoption('--build-base-image')
112118
if build_base_image:
113119
return image_utils.build_base_image(framework_name=docker_base_name,
114120
framework_version=framework_version,
121+
py_version=py_version,
115122
base_image_tag=tag,
116123
processor=processor,
117124
cwd=os.path.join(dir_path, '..'))
@@ -120,11 +127,12 @@ def fixture_build_base_image(request, framework_version, processor, tag, docker_
120127

121128

122129
@pytest.fixture(scope='session', name='build_image', autouse=True)
123-
def fixture_build_image(request, framework_version, processor, tag, docker_base_name):
130+
def fixture_build_image(request, framework_version, py_version, processor, tag, docker_base_name):
124131
build_image = request.config.getoption('--build-image')
125132
if build_image:
126133
return image_utils.build_image(framework_name=docker_base_name,
127134
framework_version=framework_version,
135+
py_version=py_version,
128136
processor=processor,
129137
tag=tag,
130138
cwd=os.path.join(dir_path, '..'))

‎test/utils/image_utils.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,19 +20,21 @@
2020
END_COLOR = '\033[0m'
2121

2222

23-
def build_base_image(framework_name, framework_version, processor, base_image_tag, cwd='.'):
23+
def build_base_image(framework_name, framework_version, py_version,
24+
processor, base_image_tag, cwd='.'):
2425
base_image_uri = get_base_image_uri(framework_name, base_image_tag)
2526

2627
dockerfile_location = os.path.join('docker', framework_version, 'base',
2728
'Dockerfile.{}'.format(processor))
2829

2930
subprocess.check_call(['docker', 'build', '-t', base_image_uri,
30-
'-f', dockerfile_location, cwd], cwd=cwd)
31+
'-f', dockerfile_location, '--build-arg',
32+
'py_version={}'.format(py_version[-1]), cwd], cwd=cwd)
3133
print('created image {}'.format(base_image_uri))
3234
return base_image_uri
3335

3436

35-
def build_image(framework_name, framework_version, processor, tag, cwd='.'):
37+
def build_image(framework_name, framework_version, py_version, processor, tag, cwd='.'):
3638
_check_call('python setup.py bdist_wheel')
3739

3840
image_uri = get_image_uri(framework_name, tag)
@@ -41,7 +43,8 @@ def build_image(framework_name, framework_version, processor, tag, cwd='.'):
4143
'Dockerfile.{}'.format(processor))
4244

4345
subprocess.check_call(
44-
['docker', 'build', '-t', image_uri, '-f', dockerfile_location, cwd], cwd=cwd)
46+
['docker', 'build', '-t', image_uri, '-f', dockerfile_location, '--build-arg',
47+
'py_version={}'.format(py_version[-1]), cwd], cwd=cwd)
4548
print('created image {}'.format(image_uri))
4649
return image_uri
4750

0 commit comments

Comments
 (0)