Enable Multi-Output Export (#2179)

* Enable Multi-Output Export

* Add test

* Update examples and lint

* fix padding

* test ops

* dummy commit to rerun test

* revert cuda lint

* Enforce tuple/list of tensors

* subscripted generics

* put back webgpu test

* Re-enable WebGPU Efficientnet test
This commit is contained in:
Akshay Kashyap
2023-10-30 21:42:26 -04:00
committed by GitHub
parent a7439af786
commit 018bd29e37
6 changed files with 63 additions and 28 deletions

View File

@@ -243,6 +243,10 @@ jobs:
run: WEBGPU=1 WGPU_BACKEND_TYPE=Metal python -m pytest -n=auto test/test_dtype.py
- name: Build WEBGPU Efficientnet
run: WEBGPU=1 WGPU_BACKEND_TYPE=Metal python -m examples.compile_efficientnet
- name: Install Puppeteer
run: npm install puppeteer
- name: Run WEBGPU Efficientnet
run: node test/test_webgpu.js
tests:
strategy: