Преглед на файлове

feat(device_capabilities.py): add support for NVIDIA RTX 4000 ADA generation device capabilities

LIPERE Benjamin преди 9 месеца
родител
ревизия
ade9db4dfd
променени са 1 файла, в които са добавени 2 реда и са изтрити 0 реда
  1. 2 0
      exo/topology/device_capabilities.py

+ 2 - 0
exo/topology/device_capabilities.py

@@ -97,6 +97,8 @@ CHIP_FLOPS = {
   "NVIDIA RTX A4500": DeviceFlops(fp32=23.65*TFLOPS, fp16=23.65*TFLOPS, int8=94.6*TFLOPS),
   "NVIDIA RTX A5000": DeviceFlops(fp32=27.8*TFLOPS, fp16=27.8*TFLOPS, int8=111.2*TFLOPS),
   "NVIDIA RTX A6000": DeviceFlops(fp32=38.71*TFLOPS, fp16=38.71*TFLOPS, int8=154.84*TFLOPS),
+  # NVIDIA Ada Lovelace Architecture-Based
+  "NVIDIA RTX 4000 ADA GENERATION": DeviceFlops(fp32=26.7*TFLOPS, fp16=26.7*TFLOPS, int8=258.0*TFLOPS),
   # Common Server GPUs
   "NVIDIA A40 48GB PCIE": DeviceFlops(fp32=37.4*TFLOPS, fp16=149.7*TFLOPS, int8=299.3*TFLOPS),
   "NVIDIA A100 40GB PCIE": DeviceFlops(fp32=19.5*TFLOPS, fp16=312.0*TFLOPS, int8=624.0*TFLOPS),