diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index e382cca8c1..1e66c638a2 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -130,7 +130,7 @@ jobs: with: python-version: 3.8 - name: Install Dependencies - run: pip install -e '.[gpu,testing]' --extra-index-url https://download.pytorch.org/whl/cpu + run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu - name: Run Optimizer Test (OPT 2 and 3) run: | PYTHONPATH="." OPT=2 GPU=1 python test/external/external_test_opt.py @@ -158,7 +158,7 @@ jobs: with: python-version: 3.8 - name: Install Dependencies - run: pip install -e '.[gpu,testing]' --extra-index-url https://download.pytorch.org/whl/cpu + run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu - name: Test GPU IMAGE ops run: | GPU=1 IMAGE=1 python3 test/test_ops.py diff --git a/setup.py b/setup.py index 9a5de62d30..3aef185bc1 100644 --- a/setup.py +++ b/setup.py @@ -19,10 +19,9 @@ setup(name='tinygrad', "Programming Language :: Python :: 3", "License :: OSI Approved :: MIT License" ], - install_requires=['numpy', 'requests', 'pillow', 'tqdm', 'networkx'], + install_requires=['numpy', 'requests', 'pillow', 'tqdm', 'networkx', 'pyopencl'], python_requires='>=3.8', extras_require={ - 'gpu': ["pyopencl"], 'llvm': ["llvmlite"], 'cuda': ["pycuda"], 'triton': ["triton>=2.0.0.dev20221202"],