wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-08 21:34:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Core] Accelerator type API (#10561)

Browse Source
This commit is contained in:
Alex Wu
2020-09-06 20:58:40 -07:00
committed by GitHub
parent a699f6a4d8
commit d6a9f0e2e4
10 changed files with 133 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/util/accelerators/__init__.py
+6
View File
@@ -0,0 +1,6 @@
from ray.util.accelerators.accelerators import NVIDIA_TESLA_V100
__all__ = [
"NVIDIA_TESLA_V100", "NVIDIA_TESLA_P100", "NVIDIA_TESLA_T4",
"NVIDIA_TESLA_P4", "NVIDIA_TESLA_K80"
]
python/ray/util/accelerators/accelerators.py
+5
View File
@@ -0,0 +1,5 @@
NVIDIA_TESLA_V100 = "V100"
NVIDIA_TESLA_P100 = "P100"
NVIDIA_TESLA_T4 = "T4"
NVIDIA_TESLA_P4 = "P4"
NVIDIA_TESLA_K80 = "K80"