Files
tinygrad/test/test_multitensor.py
Yixiang Gao 259bf9bffc add multigpu test for RMSNorm (#3056)
* need all gather

* add two multigpu test scenarios for RMSNorm
2024-01-09 09:52:51 -08:00

7.6 KiB