OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
ae01f2b61b
pytorch/torch/testing/_internal
History
Nikita Shulga ae01f2b61b Extend CPU implementation of MSELoss to BF16 (#139959) 
It's strange that it has not been implemented for the type yet

Pull Request resolved: https://github.com/pytorch/pytorch/pull/139959
Approved by: https://github.com/jgong5, https://github.com/janeyx99
ghstack dependencies: #139961
2024-11-07 23:50:15 +00:00
..
codegen
data
distributed [DTensor][Test] Fix gloo backend failure when eager_init is turned on (#139097) 2024-10-29 00:04:06 +00:00
generated
opinfo Disable foreach tests for complex128 internally (#139649) 2024-11-04 23:24:47 +00:00
optests Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)" 2024-11-05 23:10:38 +00:00
test_module
__init__.py
autocast_test_lists.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
autograd_function_db.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
check_kernel_launches.py
common_cuda.py Make sure all SDPA tests are ran with tensor cores enabled (#135592) 2024-10-29 20:53:10 +00:00
common_device_type.py [ROCm] re-enable flex attention UTs (#139632) 2024-11-06 12:49:44 +00:00
common_dist_composable.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
common_distributed.py Ensure TORCH_TRACE is run for Dynamo/Distributed tests (#139786) 2024-11-07 01:58:05 +00:00
common_dtype.py [redo] Fp8 support for item() with cuda, index_select, and fill_ cpu (#137341) 2024-10-07 00:58:51 +00:00
common_fsdp.py [CI] Add Compiled DDP / Compiled FSDP2 / compute-comm reordering tests to test_inductor_distributed (#138178) 2024-10-20 19:38:18 +00:00
common_jit.py
common_methods_invocations.py Extend CPU implementation of MSELoss to BF16 (#139959) 2024-11-07 23:50:15 +00:00
common_mkldnn.py
common_modules.py [MPS][Perf] Dispatch to SDP-math-mps for non-contig Tensors (#139791) 2024-11-06 16:25:39 +00:00
common_nn.py
common_optimizers.py Add Support for Tracking Parameter Names (named_parameters) in Optimizer State Dict (#134107) 2024-10-14 19:24:44 +00:00
common_pruning.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
common_quantization.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
common_quantized.py [3/N] Don't skip ASAN on some tests (#139058) 2024-10-28 23:57:23 +00:00
common_subclass.py Fix wrapper subclass serialization with custom sizes / strides (#137030) 2024-10-02 18:55:03 +00:00
common_utils.py [profiler][UT] instantiate profiler UTs for devices and enable UTs for xpu profiler (#134316) 2024-11-05 05:46:13 +00:00
composite_compliance.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
custom_op_db.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
custom_tensor.py
dist_utils.py
dynamo_test_failures.py
fake_config_module.py Add type annotations to Configs (#139833) 2024-11-07 03:49:09 +00:00
hop_db.py [hop-db] Import invoke_subgraph to avoid Dynamo error on mac (#140038) 2024-11-07 22:36:57 +00:00
hypothesis_utils.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
inductor_utils.py [Inductor UT] Generalize newly introduced inductor UTs for intel GPU (Part 3) (#136947) 2024-10-12 13:21:20 +00:00
jit_metaprogramming_utils.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
jit_utils.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
logging_tensor.py
logging_utils.py
quantization_torch_package_models.py
static_module.py
subclasses.py [aotd] coerce_same_metadata_as_tangent with expected_type for e.g.AsyncCollectiveTensor (#139095) 2024-11-07 16:24:48 +00:00
torchbind_impls.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
triton_utils.py Add host-side Triton TMA support to Dynamo (#137677) 2024-10-16 02:18:48 +00:00
two_tensor.py Fix tensor subclass + dynamic shapes in torch.compile + aot autograd (#125941) 2024-10-28 21:58:59 +00:00