wassname/ray
1
0 Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-10-29 13:30:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix] accelerator_types (#10725)

Browse Source 
* .

* .
This commit is contained in:
Alex Wu authored and Barak Michener committed 2020-09-14 17:09:28 +00:00
1 parent c4aaeab256
commit 72e19ede28
1 file changed
+2 -1
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",