in ossci/caffe2/Images.groovy [96:121]
static {
// By default populated by all dockerBaseImages pointing to themselves
for (String baseImage : dockerBaseImages) {
baseImageOf.put(baseImage, baseImage);
}
//////////////////////////////////////////////////////////////////////////
// Additional builds that re-use dockerBaseImages below
// Aten builds
//baseImageOf.put("py2-mkl-aten-ubuntu16.04", "py2-mkl-ubuntu16.04")
//baseImageOf.put("py2-cuda8.0-cudnn7-aten-ubuntu16.04", "py2-cuda8.0-cudnn7-ubuntu16.04")
//baseImageOf.put("py2-cuda9.0-cudnn7-aten-ubuntu16.04", "py2-cuda9.0-cudnn7-ubuntu16.04")
//// environment that is used to run pytorch->onnx->caffe2 integration tests
//baseImageOf.put("onnx-py2-gcc5-ubuntu16.04", "py2-gcc5-ubuntu16.04")
// Verify that all docker images (values) in the map are valid
for (String baseImage : baseImageOf.values()) {
assert baseImage in dockerBaseImages
}
// Verify that all base images are in the map
for (String baseImage : dockerBaseImages) {
assert baseImage in baseImageOf.keySet()
}
}