trunk/25aff58393f77a4e995620caaa35196357f3c1de: [Test]Make test/nn/test_parametrization.py device generic (#184661)
🔒
https://github.com
«Remove only_for=("cpu", "cuda") restriction from
TestNNParametrizationDevice to enable tests on all accelerator devices.
test_weight_norm_parametrization uses only device-agnostic operations:
torch.randn(..., device=dev...»
Automatische Weiterleitung...
1.5s