wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-30 01:57:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix] accelerator_types (#10725)

Browse Source 
* .

* .
This commit is contained in:
Alex Wu
2020-09-11 12:43:43 -07:00
committed by Barak Michener
parent c4aaeab256
commit 72e19ede28
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/util/accelerators/__init__.py
+2 -1
View File
@@ -1,4 +1,5 @@
from ray.util.accelerators.accelerators import NVIDIA_TESLA_V100
from ray.util.accelerators.accelerators import NVIDIA_TESLA_V100, \
NVIDIA_TESLA_P100, NVIDIA_TESLA_T4, NVIDIA_TESLA_P4, NVIDIA_TESLA_K80
__all__ = [
"NVIDIA_TESLA_V100", "NVIDIA_TESLA_P100", "NVIDIA_TESLA_T4",