youkaichao bc6e42a9b1 [hardware][rocm] allow rocm to override default env var (#7926)

2024-08-27 19:50:06 -07:00

..

__init__.py

[TPU] Avoid initializing TPU runtime in is_tpu (#7763)

2024-08-21 21:31:49 -07:00

cuda.py

[cuda][misc] error on empty CUDA_VISIBLE_DEVICES (#7924)

2024-08-27 12:06:11 -07:00

interface.py

[misc] use nvml to get consistent device name (#7582)

2024-08-16 21:15:13 -07:00

rocm.py

[hardware][rocm] allow rocm to override default env var (#7926)

2024-08-27 19:50:06 -07:00

tpu.py

[Misc] Add a wrapper for torch.inference_mode (#6618)

2024-07-21 18:43:11 -07:00