From bfd4f4e35cde7c582f8b3283fec65f02bb139fc0 Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Mon, 9 Jan 2023 12:41:44 -0800
Subject: [PATCH] testdocker

---
 .github/workflows/test.yml | 9 +++++++++
 setup.py                   | 2 +-
 test/Dockerfile            | 4 ++++
 3 files changed, 14 insertions(+), 1 deletion(-)
 create mode 100644 test/Dockerfile

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index fa35006c1e..1e67f43709 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -172,3 +172,12 @@ jobs:
       run: pip install -e '.[testing]'
     - name: Run mypy
       run: mypy tinygrad/
+
+  testdocker:
+    name: Docker Test
+    runs-on: ubuntu-latest
+    steps:
+    - name: Build Docker
+      run: docker build -t tinygrad -f test/Dockerfile .
+    - name: Test Docker
+      run: docker run -it --rm tinygrad /usr/bin/env python3 -c "from tinygrad.tensor import Tensor; print(Tensor.eye(3).numpy())"
\ No newline at end of file
diff --git a/setup.py b/setup.py
index f025e8d438..cd4897230f 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@ setup(name='tinygrad',
       license='MIT',
       long_description=long_description,
       long_description_content_type='text/markdown',
-      packages = ['tinygrad'],
+      packages = ['tinygrad', 'tinygrad.llops', 'tinygrad.nn'],
       classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: MIT License"
diff --git a/test/Dockerfile b/test/Dockerfile
new file mode 100644
index 0000000000..7bf10fc6a1
--- /dev/null
+++ b/test/Dockerfile
@@ -0,0 +1,4 @@
+FROM ubuntu:20.04
+RUN apt-get update 
+RUN apt-get install -y python3-pip git
+RUN pip3 install git+https://github.com/geohot/tinygrad.git