Merge pull request #3 from ewanwm/Fix_PyTorch_test_build

Fix py torch test build

.github/workflows/cmake-single-platform.yml

@@ -22,17 +22,23 @@ jobs:
 
     steps:
     # this Action should follow steps to set up Python build environment
+    - uses: actions/checkout@v4
+
+    - name: Set Flags
+      run: export USE_CUDA=0
+
+    - name: Install Protobuf
+      run: sudo apt install protobuf-compiler
+
     - name: Install Python dependencies
       uses: py-actions/py-dependency-install@v4
       with:
         path: "PyTorch_requirements.txt"
 
-    - uses: actions/checkout@v4
-
     - name: Configure CMake
       # Configure CMake in a 'build' subdirectory. `CMAKE_BUILD_TYPE` is only required if you are using a single-configuration generator such as make.
       # See https://cmake.org/cmake/help/latest/variable/CMAKE_BUILD_TYPE.html?highlight=cmake_build_type
-      run: cmake -B ${{github.workspace}}/build -DCMAKE_BUILD_TYPE=${{env.BUILD_TYPE}}
+      run: cmake -DCMAKE_PREFIX_PATH=`python3 -c 'import torch;print(torch.utils.cmake_prefix_path)'` -B ${{github.workspace}}/build -DCMAKE_BUILD_TYPE=${{env.BUILD_TYPE}}
 
     - name: Build
       # Build your program with the given configuration

PyTorch_requirements.txt

@@ -1,18 +1,21 @@
-attrs==23.2.0+computecanada
+attrs==23.2.0
 filelock==3.15.4
-jinja2==3.1.4+computecanada
-jsonschema-specifications==2023.12.1+computecanada
-MarkupSafe==2.1.3+computecanada
-mpmath==1.3.0+computecanada
-networkx==3.2.1+computecanada
-platformdirs==4.2.2+computecanada
-pyzmq==25.1.1+computecanada
-referencing==0.35.1+computecanada
-rpds-py==0.10.0+computecanada
-six==1.16.0+computecanada
-sympy==1.12.1+computecanada
-torch==2.0.1+computecanada
-tornado==6.3.3+computecanada
-traitlets==5.14.3+computecanada
-typing-extensions==4.12.2+computecanada
+jinja2==3.1.4
+jsonschema-specifications==2023.12.1
+MarkupSafe==2.1.3
+mpmath==1.3.0
+networkx==3.2.1
+platformdirs==4.2.2
+pyzmq==25.1.1
+referencing==0.35.1
+rpds-py==0.10.0
+six==1.16.0
+sympy==1.12.1
+
+--find-links https://download.pytorch.org/whl/torch_stable.html
+torch==2.0.1+cpu
+
+tornado==6.3.3
+traitlets==5.14.3
+typing-extensions==4.12.2
 zipp==3.19.2

nuTens/propagator/CMakeLists.txt

@@ -0,0 +1 @@
+