testdocker

.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())"

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"

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