tinygrad/extra/torch_backend
Xingyu a21369d039
Enhance tensor random functions with dtype support (#10214)
* Enhance tensor random functions with dtype support
- Updated `aten.uniform_` and `aten.normal_` to include dtype parameter in backend.py
- Added unit tests for uniform and normal tensor generation with specific dtypes in test.py

* Refactor test name for clarity
- Renamed `test_normal_dtype` to `test_normal` in `extra/torch_backend/test.py`
- Aims to improve readability and better reflect the test's purpose
2025-05-08 20:48:07 -04:00
..
.gitignore torch backend works for ResNet-18 (#9200) 2025-02-22 22:16:23 +08:00
backend.py Enhance tensor random functions with dtype support (#10214) 2025-05-08 20:48:07 -04:00
backend2.py torch and numpy dtype interop [pr] (#9224) 2025-02-24 18:26:49 +08:00
example.py run some tests from torch [pr] (#9252) 2025-02-26 15:42:22 +08:00
test.py Enhance tensor random functions with dtype support (#10214) 2025-05-08 20:48:07 -04:00
test_compile.py start torch.compile support (#9279) 2025-02-27 10:29:51 +08:00
test_inplace.py better fix inplace detach (#9557) 2025-03-24 22:50:28 +08:00
test_multigpu.py torch backend multigpu - add devices and tests (#9414) 2025-03-12 11:33:11 +08:00
torch_tests.py rig up torch's testing framework [pr] (#9254) 2025-02-26 18:46:22 +08:00
wrapped_tensor.cpp torch backend: partial fix for strided related test fails (#9642) 2025-03-31 05:45:18 -04:00