static

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()
    }
  }