maybe now

This commit is contained in:
Steven Anderson
2023-08-01 02:56:37 -04:00
parent b9d12a5871
commit 0b06e231cd
2 changed files with 3 additions and 5 deletions
+3 -3
View File
@@ -60,14 +60,14 @@ jobs:
- name: Build Docker Image
run: |
docker buildx build \
--cache-from=type=local,src=/tmp/.buildx-cache \
--cache-to=type=local,dest=/tmp/.buildx-cache \
--cache-from=type=local,src=~/.cache/buildx-cache \
--cache-to=type=local,dest=~/.cache/buildx-cache \
--platform linux/arm64 \
--load \
-t tinygrad-arm64 \
-f test/extra/arm64_dockerfile .
- name: Test Ops
run: docker run -e ARM64=1 -e CLANG=1 -e FORWARD_ONLY=1 tinygrad-arm64 python3 -m pytest -n=auto test/test_ops.py
run: docker run -e ARM64=1 -e CLANG=1 -e FORWARD_ONLY=1 tinygrad-arm64 /usr/bin/env python3 -m pytest -n=auto -s test/test_ops.py
testcpuimagenet:
name: CPU and ImageNet to C Tests
-2
View File
@@ -2,8 +2,6 @@ FROM ubuntu:20.04
WORKDIR /app
ADD . /app
RUN apt-get update && \
apt-get install -y --no-install-recommends clang g++ ocl-icd-opencl-dev python3.8 python3-pip python3-dev && \
rm -rf /var/lib/apt/lists/*