From d31e220cbff75c532acdd4fa423c6a57076ac8cf Mon Sep 17 00:00:00 2001
From: chenyu <chenyu@fastmail.com>
Date: Wed, 24 Apr 2024 23:34:34 -0400
Subject: [PATCH] add mlperf-logging to setup.py mlperf (#4289)

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index ee3c96eb1c..e673c0025f 100644
--- a/setup.py
+++ b/setup.py
@@ -36,6 +36,7 @@ setup(name='tinygrad',
             "ruff",
             "types-tqdm",
         ],
+        'mlperf': ["mlperf-logging @ git+https://github.com/mlperf/logging.git@4.0.0-rc2"],
         'testing': [
             "torch",
             "pillow",