pytorch/torch/testing/_internal
Xiang Gao b822aba8ec Enable BFloat support for gemms on arch other than ampere (#50442)
Summary:
Fixes #{issue number}

Pull Request resolved: https://github.com/pytorch/pytorch/pull/50442

Reviewed By: bdhirsh

Differential Revision: D26044981

Pulled By: mruberry

fbshipit-source-id: 65c42f2c1de8d24e4852a1b5bd8f4b1735b2230e
2021-01-26 11:07:07 -08:00
..
codegen Enable CUDA Fuser for ROCm (#45965) 2020-10-08 10:41:56 -07:00
data remediation of S205607 2020-07-17 17:19:47 -07:00
distributed Create a DDPLoggingData and expose it to python interface (#50622) 2021-01-25 15:23:07 -08:00
generated remediation of S205607 2020-07-17 17:19:47 -07:00
test_module Drop unused imports (#49972) 2021-01-13 12:26:17 -08:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
autocast_test_lists.py [PyTorch] Enable type check for autocast_test_lists (#45107) 2020-09-22 16:54:26 -07:00
common_cuda.py Enable BFloat support for gemms on arch other than ampere (#50442) 2021-01-26 11:07:07 -08:00
common_device_type.py OpInfo: Add DecorateInfo class similar to SkipInfo for decorators (#50501) 2021-01-25 04:51:04 -08:00
common_distributed.py Let TensorPipe detect peer access (#50676) 2021-01-20 08:04:51 -08:00
common_jit.py JIT Testing: Improve assertAutodiffNode error message (#50626) 2021-01-20 10:05:52 -08:00
common_methods_invocations.py Enable BFloat support for gemms on arch other than ampere (#50442) 2021-01-26 11:07:07 -08:00
common_nn.py Turn on batched grad testing for CriterionTest (#50744) 2021-01-26 07:37:15 -08:00
common_quantization.py quantization: Linear + BatchNorm1d fusion (#50748) 2021-01-20 12:59:02 -08:00
common_quantized.py [quant] Quantizable LSTM (#49671) 2020-12-30 15:21:38 -08:00
common_utils.py Rename set_deterministic to use_deterministic_algorithms (#49904) 2021-01-22 11:27:07 -08:00
dist_utils.py [RPC profiling] Extend RPC profiling to support async function execution over RPC. (#44664) 2020-09-25 13:19:26 -07:00
expecttest.py Fix test_dispatch.py when running with TORCH_SHOW_CPP_STACKTRACES=1 (#50509) 2021-01-20 10:15:01 -08:00
hypothesis_utils.py [quant] conv_transpose1d_prepack / conv_transpose1d_unpack (#40360) 2020-09-11 04:12:28 -07:00
jit_metaprogramming_utils.py Improve complex support in common_nn test machinery (#50593) 2021-01-22 09:44:45 -08:00
jit_utils.py [TensorExpr] Use wider type for scalars (#50774) 2021-01-20 15:12:27 -08:00
mypy_wrapper.py [reland] Refactor mypy configs list into editor-friendly wrapper (#50826) 2021-01-26 09:04:14 -08:00
te_utils.py Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00