From 615ad96c49cad1079643f2745f66679e3b6db4bd Mon Sep 17 00:00:00 2001
From: Guilherme Caponetto <638737+caponetto@users.noreply.github.com>
Date: Wed, 17 Jul 2024 07:18:37 -0300
Subject: [PATCH] Add Tensorflow/PyTorch version compatibility

---
 ci/version-compatibility/pytorch.json    | 184 +++++++
 ci/version-compatibility/tensorflow.json | 596 +++++++++++++++++++++++
 2 files changed, 780 insertions(+)
 create mode 100644 ci/version-compatibility/pytorch.json
 create mode 100644 ci/version-compatibility/tensorflow.json

diff --git a/ci/version-compatibility/pytorch.json b/ci/version-compatibility/pytorch.json
new file mode 100644
index 000000000..1149c2740
--- /dev/null
+++ b/ci/version-compatibility/pytorch.json
@@ -0,0 +1,184 @@
+{
+  "2.3.1": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.18.1",
+    "torchaudio": "2.3.0",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "6.0"
+    }
+  },
+  "2.3.0": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.18.0",
+    "torchaudio": "2.3.0",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "6.0"
+    }
+  },
+  "2.2.2": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.17.2",
+    "torchaudio": "2.2.2",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "5.7"
+    }
+  },
+  "2.2.1": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.17.1",
+    "torchaudio": "2.2.1",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "5.7"
+    }
+  },
+  "2.2.0": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.17.0",
+    "torchaudio": "2.2.0",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "5.7"
+    }
+  },
+  "2.1.2": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.16.2",
+    "torchaudio": "2.1.2",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "5.6"
+    }
+  },
+  "2.1.1": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.16.1",
+    "torchaudio": "2.1.1",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "5.6"
+    }
+  },
+  "2.1.0": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.16.0",
+    "torchaudio": "2.1.0",
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7.0.84",
+      "rocm": "5.6"
+    }
+  },
+  "2.0.1": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.15.2",
+    "torchaudio": "2.0.2",
+    "gpu": {
+      "cuda": "11.7",
+      "cudnn": "8.5.0.96",
+      "rocm": "5.4"
+    }
+  },
+  "2.0.0": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "torchvision": "0.15.0",
+    "torchaudio": "2.0.0",
+    "gpu": {
+      "cuda": "11.7",
+      "cudnn": "8.5.0.96",
+      "rocm": "5.4"
+    }
+  },
+  "1.13.1": {
+    "python": {
+      "min": "3.7",
+      "max": "3.10"
+    },
+    "torchvision": "0.14.1",
+    "torchaudio": "0.13.1",
+    "gpu": {
+      "cuda": "11.6",
+      "cudnn": "8.3.2.44",
+      "rocm": "5.2"
+    }
+  },
+  "1.13.0": {
+    "python": {
+      "min": "3.7",
+      "max": "3.10"
+    },
+    "torchvision": "0.14.0",
+    "torchaudio": "0.13.0",
+    "gpu": {
+      "cuda": "11.6",
+      "cudnn": "8.3.2.44",
+      "rocm": "5.2"
+    }
+  },
+  "1.12.1": {
+    "python": {
+      "min": "3.7",
+      "max": "3.10"
+    },
+    "torchvision": "0.13.1",
+    "torchaudio": "0.12.1",
+    "gpu": {
+      "cuda": "11.3",
+      "cudnn": "8.3.2.44",
+      "rocm": "5.0"
+    }
+  },
+  "1.12.0": {
+    "python": {
+      "min": "3.7",
+      "max": "3.10"
+    },
+    "torchvision": "0.13.0",
+    "torchaudio": "0.12.0",
+    "gpu": {
+      "cuda": "11.3",
+      "cudnn": "8.3.2.44",
+      "rocm": "5.0"
+    }
+  }
+}
diff --git a/ci/version-compatibility/tensorflow.json b/ci/version-compatibility/tensorflow.json
new file mode 100644
index 000000000..05122b35a
--- /dev/null
+++ b/ci/version-compatibility/tensorflow.json
@@ -0,0 +1,596 @@
+{
+  "2.16.1": {
+    "python": {
+      "min": "3.9",
+      "max": "3.12"
+    },
+    "compiler": {
+      "type": "clang",
+      "version": "17.0.6"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "6.5.0"
+    },
+    "gpu": {
+      "cuda": "12.3",
+      "cudnn": "8.9"
+    }
+  },
+  "2.15.0": {
+    "python": {
+      "min": "3.9",
+      "max": "3.11"
+    },
+    "compiler": {
+      "type": "clang",
+      "version": "16.0.0"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "6.1.0"
+    },
+    "gpu": {
+      "cuda": "12.2",
+      "cudnn": "8.9"
+    }
+  },
+  "2.14.0": {
+    "python": {
+      "min": "3.9",
+      "max": "3.11"
+    },
+    "compiler": {
+      "type": "clang",
+      "version": "16.0.0"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "6.1.0"
+    },
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.7"
+    }
+  },
+  "2.13.0": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "compiler": {
+      "type": "clang",
+      "version": "16.0.0"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "5.3.0"
+    },
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.6"
+    }
+  },
+  "2.12.0": {
+    "python": {
+      "min": "3.8",
+      "max": "3.11"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "9.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "5.3.0"
+    },
+    "gpu": {
+      "cuda": "11.8",
+      "cudnn": "8.6"
+    }
+  },
+  "2.11.0": {
+    "python": {
+      "min": "3.7",
+      "max": "3.10"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "9.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "5.3.0"
+    },
+    "gpu": {
+      "cuda": "11.2",
+      "cudnn": "8.1"
+    }
+  },
+  "2.10.0": {
+    "python": {
+      "min": "3.7",
+      "max": "3.10"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "9.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "5.1.1"
+    },
+    "gpu": {
+      "cuda": "11.2",
+      "cudnn": "8.1"
+    }
+  },
+  "2.9.0": {
+    "python": {
+      "min": "3.7",
+      "max": "3.10"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "9.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "5.0.0"
+    },
+    "gpu": {
+      "cuda": "11.2",
+      "cudnn": "8.1"
+    }
+  },
+  "2.8.0": {
+    "python": {
+      "min": "3.7",
+      "max": "3.10"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "7.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "4.2.1"
+    },
+    "gpu": {
+      "cuda": "11.2",
+      "cudnn": "8.1"
+    }
+  },
+  "2.7.0": {
+    "python": {
+      "min": "3.7",
+      "max": "3.9"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "7.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "3.7.2"
+    },
+    "gpu": {
+      "cuda": "11.2",
+      "cudnn": "8.1"
+    }
+  },
+  "2.6.0": {
+    "python": {
+      "min": "3.6",
+      "max": "3.9"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "7.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "3.7.2"
+    },
+    "gpu": {
+      "cuda": "11.2",
+      "cudnn": "8.1"
+    }
+  },
+  "2.5.0": {
+    "python": {
+      "min": "3.6",
+      "max": "3.9"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "7.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "3.7.2"
+    },
+    "gpu": {
+      "cuda": "11.2",
+      "cudnn": "8.1"
+    }
+  },
+  "2.4.0": {
+    "python": {
+      "min": "3.6",
+      "max": "3.8"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "7.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "3.1.0"
+    },
+    "gpu": {
+      "cuda": "11.0",
+      "cudnn": "8.0"
+    }
+  },
+  "2.3.0": {
+    "python": {
+      "min": "3.5",
+      "max": "3.8"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "7.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "3.1.0"
+    },
+    "gpu": {
+      "cuda": "10.1",
+      "cudnn": "7.6"
+    }
+  },
+  "2.2.0": {
+    "python": {
+      "min": "3.5",
+      "max": "3.8"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "7.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "2.0.0"
+    },
+    "gpu": {
+      "cuda": "10.1",
+      "cudnn": "7.6"
+    }
+  },
+  "2.1.0": {
+    "python": {
+      "min": ["2.7", "3.5"],
+      "max": "3.7"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "7.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.27.1"
+    },
+    "gpu": {
+      "cuda": "10.1",
+      "cudnn": "7.6"
+    }
+  },
+  "2.0.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.7"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "7.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.26.1"
+    },
+    "gpu": {
+      "cuda": "10.0",
+      "cudnn": "7.4"
+    }
+  },
+  "1.15.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.7"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "7.3.1"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.26.1"
+    },
+    "gpu": {
+      "cuda": "10.0",
+      "cudnn": "7.4"
+    }
+  },
+  "1.14.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.7"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.24.1"
+    },
+    "gpu": {
+      "cuda": "10.0",
+      "cudnn": "7.4"
+    }
+  },
+  "1.13.1": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.7"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.19.2"
+    },
+    "gpu": {
+      "cuda": "10.0",
+      "cudnn": "7.4"
+    }
+  },
+  "1.12.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.15.0"
+    },
+    "gpu": {
+      "cuda": "9",
+      "cudnn": "7"
+    }
+  },
+  "1.11.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.15.0"
+    },
+    "gpu": {
+      "cuda": "9",
+      "cudnn": "7"
+    }
+  },
+  "1.10.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.15.0"
+    },
+    "gpu": {
+      "cuda": "9",
+      "cudnn": "7"
+    }
+  },
+  "1.9.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.11.0"
+    },
+    "gpu": {
+      "cuda": "9",
+      "cudnn": "7"
+    }
+  },
+  "1.8.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.10.0"
+    },
+    "gpu": {
+      "cuda": "9",
+      "cudnn": "7"
+    }
+  },
+  "1.7.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.9.0"
+    },
+    "gpu": {
+      "cuda": "9",
+      "cudnn": "7"
+    }
+  },
+  "1.6.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.9.0"
+    },
+    "gpu": {
+      "cuda": "9",
+      "cudnn": "7"
+    }
+  },
+  "1.5.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.8.0"
+    },
+    "gpu": {
+      "cuda": "9",
+      "cudnn": "7"
+    }
+  },
+  "1.4.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.5.4"
+    },
+    "gpu": {
+      "cuda": "8",
+      "cudnn": "6"
+    }
+  },
+  "1.3.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.4.5"
+    },
+    "gpu": {
+      "cuda": "8",
+      "cudnn": "6"
+    }
+  },
+  "1.2.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.4.5"
+    },
+    "gpu": {
+      "cuda": "8",
+      "cudnn": "5.1"
+    }
+  },
+  "1.1.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.4.2"
+    },
+    "gpu": {
+      "cuda": "8",
+      "cudnn": "5.1"
+    }
+  },
+  "1.0.0": {
+    "python": {
+      "min": ["2.7", "3.3"],
+      "max": "3.6"
+    },
+    "compiler": {
+      "type": "gcc",
+      "version": "4.8"
+    },
+    "build_tools": {
+      "name": "bazel",
+      "version": "0.4.2"
+    },
+    "gpu": {
+      "cuda": "8",
+      "cudnn": "5.1"
+    }
+  }
+}