pytorch/torch/testing/_internal
vasiliy 605392bd06 add float8 types to LoggingTensor (#141385)
Summary:

float8 dtypes were missing from this map, adding

Test Plan:

CI, and unbreaks debugging in torchao

If there is an existing test I can add this to - lmk

Reviewers:

Subscribers:

Tasks:

Tags:

Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/141385
Approved by: https://github.com/soulitzer
2024-11-26 23:39:57 +00:00
..
codegen
data
distributed [DTensorTestbase] Fix TestFunc typing issue (#141513) 2024-11-26 19:48:34 +00:00
generated
opinfo NJT unsqueeze() fixes (#141392) 2024-11-26 22:38:35 +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
autograd_function_db.py
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 General per-SampleInput xfail / skip system (#140443) 2024-11-19 23:09:38 +00:00
common_dist_composable.py
common_distributed.py Revert "[CI] Reduce distributed test timeout to 60s (#141168)" 2024-11-22 15:46:37 +00:00
common_dtype.py
common_fsdp.py Fix type-safety of torch.nn.Module instances (#141240) 2024-11-22 00:05:05 +00:00
common_jit.py
common_methods_invocations.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
common_mkldnn.py
common_modules.py [BE] Use torch.log1p(x) instead of torch.log(1+x) (#141167) 2024-11-21 00:36:20 +00:00
common_nn.py
common_optimizers.py Support tensor betas in Adam and AdamW (#134171) 2024-11-15 21:55:55 +00:00
common_pruning.py Fix type-safety of torch.nn.Module instances (#141240) 2024-11-22 00:05:05 +00:00
common_quantization.py [Intel GPU] XPUInductorQuantizer for XPU int8 recipe customization (#139578) 2024-11-26 09:44:14 +00:00
common_quantized.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
common_subclass.py
common_utils.py [Intel GPU] qconv at XPU backend (#133080) 2024-11-26 02:24:30 +00:00
composite_compliance.py
custom_op_db.py
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 [invoke_subgraph] User facing API to support arbitrary args and kwargs (#139162) 2024-11-08 03:31:19 +00:00
hypothesis_utils.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
inductor_utils.py
jit_metaprogramming_utils.py
jit_utils.py
logging_tensor.py add float8 types to LoggingTensor (#141385) 2024-11-26 23:39:57 +00:00
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
triton_utils.py [RFC] Implement caching for user defined triton kernels (#140326) 2024-11-16 02:37:16 +00:00
two_tensor.py Fix tensor subclass + dynamic shapes in torch.compile + aot autograd (#125941) 2024-10-28 21:58:59 +00:00