forked from tinygrad/tinygrad
maybe now
This commit is contained in:
@@ -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,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/*
|
||||
|
||||
Reference in New Issue
Block a user