PR #7896 [ROCm] Add initial ROCm PJRT support

[ROCm] Add initial ROCm PJRT support #7896

mmakevic-amd wants to merge 5 commits into pytorch:master from mmakevic-amd:rocm-plugin

pytorch-bot added rocm

JackCaoG requested a review from

will-cromar 1 year ago

will-cromar commented on 2024-09-05

mmakevic-amd closed this 1 year ago

Add rocm pjrt plugin

5f4b2451

Add missing options to .bazelrc

3a939151

Differentiate between rocm and cuda

94fbec51

Add configure_single_process to __init__.py

35955aed

Fix formatting

dc187452

mmakevic-amd reopened this 1 year ago

mmakevic-amd force pushed from 6b79b398 to dc187452 1 year ago

miladm requested a review from

amjames 1 year ago

miladm requested a review from

ysiraichi 1 year ago

miladm assigned

mmakevic-amd 1 year ago

miladm added xla:gpu

miladm added runtime

Reviewers

will-cromar

amjames

ysiraichi

Assignees

mmakevic-amd

Labels

xla:gpu runtime rocm

Milestone

No milestone