Update device_capabilities.py

d4e26fcbf7920663b77f04508817d9bb9a2d3559 · 2024-10-27 21:36:42 -0700 · FFAMax

Files touched

M exo/topology/device_capabilities.py

Diff

commit d4e26fcbf7920663b77f04508817d9bb9a2d3559
Author: FFAMax <ffamax@gmail.com>
Date:   Sun Oct 27 21:36:42 2024 -0700

    Update device_capabilities.py
---
 exo/topology/device_capabilities.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/exo/topology/device_capabilities.py b/exo/topology/device_capabilities.py
index a5f3597c..5c69e57d 100644
--- a/exo/topology/device_capabilities.py
+++ b/exo/topology/device_capabilities.py
@@ -94,6 +94,9 @@ CHIP_FLOPS = {
   "NVIDIA TITAN RTX": DeviceFlops(fp32=16.31*TFLOPS, fp16=32.62*TFLOPS, int8=65.24*TFLOPS),
   # GTX 10 series
   "NVIDIA GEFORCE GTX 1050 TI": DeviceFlops(fp32=2.0*TFLOPS, fp16=4.0*TFLOPS, int8=8.0*TFLOPS),
+  "NVIDIA GEFORCE GTX 1070": DeviceFlops(fp32=6.463*TFLOPS, fp16=0.101*TFLOPS, int8=25.852*TFLOPS),
+  "NVIDIA GEFORCE GTX 1080": DeviceFlops(fp32=8.873*TFLOPS, fp16=0.138*TFLOPS, int8=35.492*TFLOPS),
+  "NVIDIA GEFORCE GTX 1080 TI": DeviceFlops(fp32=11.34*TFLOPS, fp16=0.177*TFLOPS, int8=45.36*TFLOPS),
   # GTX 16 series
   "NVIDIA GeForce GTX 1660 TI": DeviceFlops(fp32=4.8*TFLOPS, fp16=9.6*TFLOPS, int8=19.2*TFLOPS),
   # QUADRO RTX Ampere series

← bc1d88d8 ignore dummy · back to Exo · Update main.py: Default timeout 90->900 dbf40d78 →