pytorch/test/nn
Pruthvi Madugundu 9ce2e02fd6 Revert "[ROCm] Remove PYTORCH_MIOPEN_SUGGEST_NHWC flag (#90725)" (#110319)
This reverts commit 66bfcd32fd.

NHWC is have perf regression on MIOpen, so reverting till the performance issue is fixed.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/110319
Approved by: https://github.com/jeffdaily, https://github.com/jithunnair-amd, https://github.com/kit1980
2023-10-03 19:14:47 +00:00
..
test_convolution.py Revert "[ROCm] Remove PYTORCH_MIOPEN_SUGGEST_NHWC flag (#90725)" (#110319) 2023-10-03 19:14:47 +00:00
test_dropout.py [test_nn] add custom device support for dropout tests、lazy_modules te… (#106609) 2023-08-11 09:14:34 +00:00
test_embedding.py Ensure fill_ works when value is a view of self (#109835) 2023-09-26 17:12:48 +00:00
test_init.py Thread generator through trunc_normal_ (#100810) 2023-05-12 01:04:08 +00:00
test_lazy_modules.py LazyBatchNorm{1-3}d support dict&set (#109015) 2023-09-12 09:09:59 +00:00
test_module_hooks.py [BE] Enable ruff's UP rules and autoformat nn/ mps/ and torch/ (#105436) 2023-07-21 07:38:46 +00:00
test_multihead_attention.py [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00
test_packed_sequence.py [BE]: Apply ruff PERF fixes to torch (#104917) 2023-07-11 20:45:21 +00:00
test_parametrization.py Back out "Reland "Make adding buffers more like adding parameters (#104069)" (#106224)" (#106743) 2023-08-08 15:27:34 +00:00
test_pooling.py Back out "[quant] Support integer implementations for adaptive_avg_pool2d (#104226)" (#110316) 2023-10-03 16:59:23 +00:00
test_pruning.py