From 6a01a987e47123eed1350e9d3f75594e0c884dab Mon Sep 17 00:00:00 2001 From: DeftDawg Date: Sat, 8 Mar 2025 04:03:59 -0500 Subject: [PATCH 1/2] Support for Steam Deck in exo --- 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 e60003f22..dcac5704e 100644 --- a/exo/topology/device_capabilities.py +++ b/exo/topology/device_capabilities.py @@ -121,6 +121,8 @@ def to_dict(self): "NVIDIA A800 80GB SXM": DeviceFlops(fp32=19.5*TFLOPS, fp16=312.0*TFLOPS, int8=624.0*TFLOPS), # ... add more devices if needed ... ### AMD GPUs + # Valve Steam Deck + "AMD Custom GPU 0405": DeviceFlops(fp32=1.638*TFLOPS, fp16=3.277*TFLOPS, int8=6.552*TFLOPS), # RX 6000 series "AMD Radeon RX 6900 XT": DeviceFlops(fp32=23.04*TFLOPS, fp16=46.08*TFLOPS, int8=92.16*TFLOPS), "AMD Radeon RX 6800 XT": DeviceFlops(fp32=20.74*TFLOPS, fp16=41.48*TFLOPS, int8=82.96*TFLOPS), From a67b0faeb6a6aa813a839b40f358c7115b44269e Mon Sep 17 00:00:00 2001 From: DeftDawg Date: Sun, 9 Mar 2025 01:06:56 -0500 Subject: [PATCH 2/2] tinygrad update to include ctl stack size fix in latest master --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 98b8357e8..d1c0d6abc 100644 --- a/setup.py +++ b/setup.py @@ -29,7 +29,7 @@ "transformers==4.46.3", "uuid==1.30", "uvloop==0.21.0", - "tinygrad @ git+https://github.com/tinygrad/tinygrad.git@ec120ce6b9ce8e4ff4b5692566a683ef240e8bc8", + "tinygrad @ git+https://github.com/tinygrad/tinygrad.git@b6fe5ab4dd11609ab8e8dd4cec9c6fa5cfe89bf7", ] extras_require = {