wehub-resource-sync
85 lines
1.8 KiB
Python
85 lines
1.8 KiB
Python
import warnings
|
|
|
|
from ray.util import tpu
|
|
from ray.util.accelerators.accelerators import (
|
|
AMD_INSTINCT_MI100,
|
|
AMD_INSTINCT_MI210,
|
|
AMD_INSTINCT_MI250,
|
|
AMD_RADEON_HD_7900,
|
|
AMD_RADEON_R9_200_HD_7900,
|
|
AWS_NEURON_CORE,
|
|
GOOGLE_TPU_V2,
|
|
GOOGLE_TPU_V3,
|
|
GOOGLE_TPU_V4,
|
|
GOOGLE_TPU_V5LITEPOD,
|
|
GOOGLE_TPU_V5P,
|
|
GOOGLE_TPU_V6E,
|
|
GOOGLE_TPU_V7X,
|
|
INTEL_GAUDI,
|
|
INTEL_MAX_1100,
|
|
INTEL_MAX_1550,
|
|
METAX_C500,
|
|
METAX_C550,
|
|
NVIDIA_A100,
|
|
NVIDIA_H100,
|
|
NVIDIA_L4,
|
|
NVIDIA_TESLA_A10G,
|
|
NVIDIA_TESLA_K80,
|
|
NVIDIA_TESLA_P4,
|
|
NVIDIA_TESLA_P100,
|
|
NVIDIA_TESLA_T4,
|
|
NVIDIA_TESLA_V100,
|
|
AMD_INSTINCT_MI250x,
|
|
AMD_INSTINCT_MI300x,
|
|
)
|
|
|
|
__all__ = [
|
|
"tpu",
|
|
"NVIDIA_TESLA_V100",
|
|
"NVIDIA_TESLA_P100",
|
|
"NVIDIA_TESLA_T4",
|
|
"NVIDIA_TESLA_P4",
|
|
"NVIDIA_TESLA_K80",
|
|
"NVIDIA_TESLA_A10G",
|
|
"NVIDIA_L4",
|
|
"NVIDIA_A100",
|
|
"NVIDIA_A100_40G",
|
|
"NVIDIA_A100_80G",
|
|
"NVIDIA_H100",
|
|
"INTEL_MAX_1550",
|
|
"INTEL_MAX_1100",
|
|
"INTEL_GAUDI",
|
|
"AMD_INSTINCT_MI100",
|
|
"AMD_INSTINCT_MI210",
|
|
"AMD_INSTINCT_MI250",
|
|
"AMD_INSTINCT_MI250x",
|
|
"AMD_INSTINCT_MI300x",
|
|
"AMD_RADEON_R9_200_HD_7900",
|
|
"AMD_RADEON_HD_7900",
|
|
"AWS_NEURON_CORE",
|
|
"GOOGLE_TPU_V2",
|
|
"GOOGLE_TPU_V3",
|
|
"GOOGLE_TPU_V4",
|
|
"GOOGLE_TPU_V5P",
|
|
"GOOGLE_TPU_V5LITEPOD",
|
|
"GOOGLE_TPU_V6E",
|
|
"GOOGLE_TPU_V7X",
|
|
"METAX_C500",
|
|
"METAX_C550",
|
|
# Deprecated
|
|
"NVIDIA_TESLA_A100",
|
|
]
|
|
|
|
|
|
def __getattr__(name: str):
|
|
if name == "NVIDIA_TESLA_A100":
|
|
from ray.util.annotations import RayDeprecationWarning
|
|
|
|
warnings.warn(
|
|
"NVIDIA_TESLA_A100 is deprecated, use NVIDIA_A100 instead.",
|
|
RayDeprecationWarning,
|
|
stacklevel=2,
|
|
)
|
|
return NVIDIA_A100
|
|
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|