[ROCm] Add initial ROCm PJRT support#7896

Open

mmakevic-amd wants to merge 5 commits intopytorch:masterfrom mmakevic-amd:rocm-plugin

+105-1

Commits on Dec 16, 2024

Add rocm pjrt plugin

mmakevic-amd
authored and
root
committed
Add missing options to .bazelrc

mmakevic-amd
authored and
root
committed
Differentiate between rocm and cuda

mmakevic-amd
authored and
root
committed
Add configure_single_process to __init__.py

mmakevic-amd
authored and
root
committed
Fix formatting

mmakevic-amd
authored and
root
committed