From faf80418b7de3412e6a89ff0e9f3dfd29fbe07f9 Mon Sep 17 00:00:00 2001 From: George Hotz <72895+geohot@users.noreply.github.com> Date: Thu, 25 May 2023 17:48:18 -0700 Subject: [PATCH] pyopencl by default since GPU is default (#802) --- .github/workflows/test.yml | 4 ++-- setup.py | 3 +-- 2 files changed, 3 insertions(+), 4 deletions(-) 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"],