From 30b72bd3d7ea303c59ce08a67ed2e2335c83a0ec Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Sun, 20 Aug 2023 12:27:57 -0700
Subject: [PATCH] add CI

---
 .github/workflows/test.yml | 24 ++++++++++++++++++++++++
 README                     |  2 ++
 2 files changed, 26 insertions(+)
 create mode 100644 .github/workflows/test.yml

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
new file mode 100644
index 0000000..dd2119f
--- /dev/null
+++ b/.github/workflows/test.yml
@@ -0,0 +1,24 @@
+name: Unit Tests
+
+on:
+  push:
+  pull_request:
+  workflow_dispatch:
+
+jobs:
+  testmnist:
+    name: Train mnist
+    runs-on: ubuntu-latest
+    timeout-minutes: 20
+
+    steps:
+    - name: Checkout Code
+      uses: actions/checkout@v3
+    - name: Set up Python 3.11
+      uses: actions/setup-python@v4
+      with:
+        python-version: 3.11
+    - name: Install Dependencies
+      run: pip install numpy
+    - name: Test EfficientNet
+      run: PYTHONPATH="." python mnist.py
diff --git a/README b/README
index bfa823c..b42d084 100644
--- a/README
+++ b/README
@@ -1,3 +1,5 @@
 teenygrad is <1000 mnist trainer
 
 It shares 90% of its code with tinygrad, so understanding teenygrad is a good step to understanding tinygrad
+
+While it supports most of tinygrad's functionality, the extra 3k lines in tinygrad get you speed and diverse backend support.
\ No newline at end of file