https://github.com/iain-s/torch-oneapi

Utils, tests and docs for using Intel oneAPI and PyTorch.
https://github.com/iain-s/torch-oneapi

Last synced: 4 days ago
JSON representation

Utils, tests and docs for using Intel oneAPI and PyTorch.

• Host: GitHub
• URL: https://github.com/iain-s/torch-oneapi
• Owner: Iain-S
• License: mit
• Created: 2024-01-12T13:59:32.000Z (10 months ago)
• Default Branch: main
• Last Pushed: 2024-01-12T17:35:14.000Z (10 months ago)
• Last Synced: 2024-04-18T06:53:40.656Z (7 months ago)
• Language: Python
• Size: 3.91 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# torch-oneapi

Utils, tests and docs for using Intel oneAPI and PyTorch.

## Files

### my_ddp.py

A simple linear regression DNN used to check whether we can run DDP on XPUs.

```bash

CCL_ZE_IPC_EXCHANGE=sockets TORCH_DEVICE=xpu mpirun -n 2 python my_ddp.py

```