u32 to f16 in tinygrad (#8074)

* f16 decompression in tinygrad

* Typing and cleanup
This commit is contained in:
Ahmed Harmouche
2024-12-06 12:00:13 +01:00
committed by GitHub
parent e37bff6c19
commit ce72fe1411
4 changed files with 32 additions and 41 deletions

View File

@@ -387,7 +387,7 @@ jobs:
WEBGPU=1 WGPU_BACKEND_TYPE=Vulkan python3 -m pytest -n=auto test/test_assign.py test/test_arange.py test/test_const_folding.py test/test_dtype.py \
test/test_dtype_alu.py test/test_conv.py test/test_conv_shapetracker.py test/test_nn.py test/test_ops.py test/test_optim.py \
test/test_jit.py test/test_randomness.py test/test_symbolic_ops.py test/test_symbolic_jit.py test/test_uops_stats.py test/test_uops.py \
test/testextra/test_export_model.py --durations=20
test/testextra/test_export_model.py test/testextra/test_f16_decompress.py --durations=20
- name: Run process replay tests
run: |
export PR_TITLE=$(jq -r .pull_request.title "$GITHUB_EVENT_PATH")