TFLOPS on GTX 1050 - cabelo@opensuse.org

60bb60f31567f17ad75dbff79d4022f7d05e4568 · 2024-10-14 15:20:46 -0300 · Alessandro de Oliveira Faria (A.K.A. CABELO)

Files touched

M exo/topology/device_capabilities.py

Diff

commit 60bb60f31567f17ad75dbff79d4022f7d05e4568
Author: Alessandro de Oliveira Faria (A.K.A. CABELO) <cabelo@opensuse.org>
Date:   Mon Oct 14 15:20:46 2024 -0300

    TFLOPS on GTX 1050 - cabelo@opensuse.org
---
 exo/topology/device_capabilities.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/exo/topology/device_capabilities.py b/exo/topology/device_capabilities.py
index 4acc1dd0..6ce55305 100644
--- a/exo/topology/device_capabilities.py
+++ b/exo/topology/device_capabilities.py
@@ -92,6 +92,8 @@ CHIP_FLOPS = {
   "NVIDIA GEFORCE RTX 2080": DeviceFlops(fp32=10.07*TFLOPS, fp16=20.14*TFLOPS, int8=40.28*TFLOPS),
   "NVIDIA GEFORCE RTX 2080 SUPER": DeviceFlops(fp32=11.15*TFLOPS, fp16=22.30*TFLOPS, int8=44.60*TFLOPS),
   "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),
   # QUADRO RTX Ampere series
   "NVIDIA RTX A2000": DeviceFlops(fp32=7.99*TFLOPS, fp16=7.99*TFLOPS, int8=31.91*TFLOPS),
   "NVIDIA RTX A4000": DeviceFlops(fp32=19.17*TFLOPS, fp16=19.17*TFLOPS, int8=76.68*TFLOPS),

← 03cbcca2 Fix gpu capabilities display issue. Also update the capabili · back to Exo · Update README.md d554313d →