Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add XPU support (duplicate #125) #209

Closed
wants to merge 26 commits into from
Closed
Changes from 1 commit
Commits
Show all changes
26 commits
Select commit Hold shift + click to select a range
aa714fe
Add MacOS GPU device option
ElliottKasoar Apr 3, 2024
6a96d49
Add XPU device option
ElliottKasoar May 6, 2024
84a4e5d
Update C++ XPU interface to handle multiple devices indices.
jatkinson1000 Oct 11, 2024
4c2fc90
Update ftorch.F90 for XPU support
ma595 Dec 20, 2024
34316e0
Add xpu python modifications to examples 1 and 2
ma595 Dec 20, 2024
d08ce7e
Add xpu modifications to fortran for example 2, init still not called
ma595 Dec 20, 2024
9d4aa86
Build example 3 if CUDA and MPI enabled
jwallwork23 Jan 7, 2025
5f4418a
Put model on CUDA device in simplenet
jwallwork23 Jan 13, 2025
b864b90
Run example 3 if it's been built
jwallwork23 Jan 13, 2025
911989b
Add missing imports for pt2ts
jwallwork23 Jan 13, 2025
4591325
More helpful output for simplenet_infer_python
jwallwork23 Jan 13, 2025
b4246c4
Fix numbering in CMakeLists for examples
jwallwork23 Jan 13, 2025
28e3778
Renaming in MultiGPU example; set up unit testing
jwallwork23 Jan 13, 2025
5fa7801
Raise error if no CUDA in example 3
jwallwork23 Jan 13, 2025
8c27dc1
Lint
jwallwork23 Jan 13, 2025
68f111a
Fix model filename passed to fortran
jwallwork23 Jan 13, 2025
3f3cd53
Do require mpi4py in Python script
jwallwork23 Jan 13, 2025
781a69f
Merge branch 'main' into add-devices-iccs
jwallwork23 Jan 24, 2025
c86fe4a
DO NOT MERGE - drop unit tests so we don't need to install pFUnit
jwallwork23 Jan 24, 2025
5c0f007
DO NOT MERGE: remove annoying compiler warnings
TomMelt Jan 24, 2025
3946312
chore: add device type to resnet example
TomMelt Jan 24, 2025
765f79a
Fix devices in pt filenames
jwallwork23 Jan 24, 2025
4504725
Merge remote-tracking branch 'origin/208_multi-gpu-build' into add-de…
jwallwork23 Jan 24, 2025
6034cb7
Convert example 3 to XPU
jwallwork23 Jan 24, 2025
d5e094b
DO NOT MERGE - turn off CI
jwallwork23 Jan 24, 2025
1ba10f5
Lint
jwallwork23 Jan 24, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update ftorch.F90 for XPU support
ma595 committed Dec 20, 2024
commit 4c2fc90c2f3de854066a597ca45a22e96eac5e6a
2 changes: 2 additions & 0 deletions src/ftorch.F90
Original file line number Diff line number Diff line change
@@ -49,6 +49,8 @@ module ftorch
enum, bind(c)
enumerator :: torch_kCPU = 0
enumerator :: torch_kCUDA = 1
enumerator :: torch_kMPS = 2
enumerator :: torch_kXPU = 3
end enum

!> Interface for directing `torch_tensor_from_array` to possible input types and ranks