Merge pull request #530 from datamol-io/remove_ipu

Remove IPU (again)

Author: WenkelF

.github/workflows/test.yml

@@ -16,8 +16,13 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.09", "3.10", "3.11"]
-        pytorch-version: ["2.0"]
+        include:
+          - python-version: "3.10"
+            pytorch-version: "2.0"
+          - python-version: "3.11"
+            pytorch-version: "2.0"
+          - python-version: "3.12"
+            pytorch-version: "2.3"
 
     runs-on: "ubuntu-latest"
     timeout-minutes: 30
@@ -55,9 +60,6 @@ jobs:
       - name: Install C++ library
         run: cd graphium/graphium_cpp && git clone https://github.com/pybind/pybind11.git && export PYTHONPATH=$PYTHONPATH:./pybind11 && python -m pip install . && cd ../..
 
-      - name: Run tests
-        run: pytest -m 'not ipu'
-
       - name: Test CLI
         run: graphium --help
 

.github/workflows/test_ipu.yml

@@ -55,15 +55,6 @@ debug/
 change_commits.sh
 graphium/features/test_new_pes.ipynb
 
-# IPU related ignores and profiler outputs
-*.a
-*.cbor
-*.capnp
-*.pop
-*.popart
-*.pop_cache
-*.popef
-*.pvti*
 
 ############ END graphium Custom GitIgnore ##############
 

.gitignore

@@ -13,7 +13,6 @@
 [![GitHub Repo stars](https://img.shields.io/github/stars/datamol-io/graphium)](https://github.com/datamol-io/graphium/stargazers)
 [![GitHub Repo stars](https://img.shields.io/github/forks/datamol-io/graphium)](https://github.com/datamol-io/graphium/network/members)
 [![test](https://github.com/datamol-io/graphium/actions/workflows/test.yml/badge.svg)](https://github.com/datamol-io/graphium/actions/workflows/test.yml)
-[![test-ipu](https://github.com/datamol-io/graphium/actions/workflows/test_ipu.yml/badge.svg)](https://github.com/datamol-io/graphium/actions/workflows/test_ipu.yml)
 [![release](https://github.com/datamol-io/graphium/actions/workflows/release.yml/badge.svg)](https://github.com/datamol-io/graphium/actions/workflows/release.yml)
 [![code-check](https://github.com/datamol-io/graphium/actions/workflows/code-check.yml/badge.svg)](https://github.com/datamol-io/graphium/actions/workflows/code-check.yml)
 [![doc](https://github.com/datamol-io/graphium/actions/workflows/doc.yml/badge.svg)](https://github.com/datamol-io/graphium/actions/workflows/doc.yml)
@@ -53,18 +52,6 @@ mamba activate graphium
 pip install --no-deps -e .
 ```
 
-### For IPU developers
-```bash
-# Install Graphcore's SDK and Graphium dependencies in a new environment called `.graphium_ipu`
-./install_ipu.sh .graphium_ipu
-```
-
-The above step needs to be done once. After that, enable the SDK and the environment as follows:
-
-```bash
-source enable_ipu.sh .graphium_ipu
-```
-
 ## Training a model
 
 To learn how to train a model, we invite you to look at the documentation, or the jupyter notebooks available [here](https://github.com/datamol-io/graphium/tree/master/docs/tutorials/model_training).

README.md

@@ -16,8 +16,3 @@ component_management:
         target: auto
         branches:
           - "!main"
-  individual_components:
-    - component_id: ipu # this is an identifier that should not be changed
-      name: ipu # this is a display name, and can be changed freely
-      paths:
-        - graphium/ipu/**

codecov.yml

@@ -18,21 +18,6 @@ mamba activate graphium
 pip install --no-deps -e .
 ```
 
-### For IPU developers
-
-Download the SDK and use pypi to create your environment:
-
-```bash
-# Install Graphcore's SDK and Graphium dependencies in a new environment called `.graphium_ipu`
-./install_ipu.sh .graphium_ipu
-```
-
-The above step needs to be done once. After that, enable the SDK and the environment as follows:
-
-```bash
-source enable_ipu.sh .graphium_ipu
-```
-
 ## Build the documentation
 
 You can build and serve the documentation locally with:

docs/api/graphium.ipu.md

@@ -42,7 +42,6 @@ Below are a list of directory and their respective documentations:
 - [data](https://github.com/datamol-io/graphium/blob/main/graphium/data/README.md)
 - [features](https://github.com/datamol-io/graphium/tree/main/graphium/features/README.md)
 - finetuning
-- [ipu](https://github.com/datamol-io/graphium/tree/main/graphium/ipu/README.md)
 - [nn](https://github.com/datamol-io/graphium/tree/main/graphium/nn/README.md)
 - [trainer](https://github.com/datamol-io/graphium/tree/main/graphium/trainer/README.md)
 - [utils](https://github.com/datamol-io/graphium/tree/main/graphium/features/README.md)
@@ -56,7 +55,7 @@ Hence, we use [hydra](https://hydra.cc/docs/intro/) to enable splitting the conf
 
 Examples of possibilities include:
 
-- Switching between accelerators (CPU, GPU and IPU)
+- Switching between accelerators (CPU, GPU)
 - Benchmarking different models on the same dataset
 - Fine-tuning a pre-trained model on a new dataset
 

docs/contribute.md

@@ -25,17 +25,6 @@ or pip:
 pip install graphium
 ```
 
-### For IPU
-```bash
-# Install Graphcore's SDK and Graphium dependencies in a new environment called `.graphium_ipu`
-./install_ipu.sh .graphium_ipu
-```
-
-The above step needs to be done once. After that, enable the SDK and the environment as follows:
-
-```bash
-source enable_ipu.sh .graphium_ipu
-```
 
 Finally, you will need to install graphium with pip
 ```bash

docs/design.md

@@ -476,7 +476,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "graphium_ipu",
+   "display_name": "graphium",
    "language": "python",
    "name": "python3"
   },