From 21158d74b3935ad579748bde18b55fcfeec368bc Mon Sep 17 00:00:00 2001
From: Dmitry Starkov <21260939+starkdmi@users.noreply.github.com>
Date: Wed, 20 Nov 2024 15:47:48 +0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=91=E2=80=8D=F0=9F=92=BB=20Update=20Ap?=
 =?UTF-8?q?ple=20Silicon=20hardware,=20add=20M4=20series=20(#1044)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Fixed:
- M2 Pro ~6.8 (FP32, 19 cores)
- M3 4.1 (FP32, 10 cores)
- M3 Pro 6.39-7.4 (FP32, 18 cores)

New:
- M4 4.6 (FP32, 10 cores)
- M4 Pro 9.2 (FP32, 20 cores)
- M4 Max 18.4 (FP32, 40 cores)
---
 packages/tasks/src/hardware.ts | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/packages/tasks/src/hardware.ts b/packages/tasks/src/hardware.ts
index 2bae4bdb6..efde8177b 100644
--- a/packages/tasks/src/hardware.ts
+++ b/packages/tasks/src/hardware.ts
@@ -452,7 +452,7 @@ export const SKUS = {
 				memory: [8, 16, 24],
 			},
 			"Apple M2 Pro": {
-				tflops: 13.6,
+				tflops: 6.8,
 				memory: [16, 24, 32],
 			},
 			"Apple M2 Max": {
@@ -464,17 +464,29 @@ export const SKUS = {
 				memory: [64, 96, 128, 192],
 			},
 			"Apple M3": {
-				tflops: 2.84,
+				tflops: 4.1,
 				memory: [8, 16, 24],
 			},
 			"Apple M3 Pro": {
-				tflops: 14,
+				tflops: 7.4,
 				memory: [18, 36],
 			},
 			"Apple M3 Max": {
 				tflops: 14.2,
 				memory: [36, 48, 64, 96, 128],
 			},
+			"Apple M4": {
+				tflops: 4.6,
+				memory: [16, 24, 32],
+			},
+			"Apple M4 Pro": {
+				tflops: 9.2,
+				memory: [24, 48],
+			},
+			"Apple M4 Max": {
+				tflops: 18.4,
+				memory: [36, 48, 64, 128],
+			},
 		},
 	},
 } satisfies Record<string, Record<string, Record<string, HardwareSpec>>>;