pytorch/test
Thomas J. Fan a7f4f80903 ENH Adds dtype to nn.functional.one_hot (#58090)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/33046
Related to https://github.com/pytorch/pytorch/issues/53785

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

Reviewed By: zou3519

Differential Revision: D28640893

Pulled By: jbschlosser

fbshipit-source-id: 3686579517ccc75beaa74f0f6d167f5e40a83fd2
2021-05-24 13:48:25 -07:00
..
backward_compatibility Revert "Striding for lists Part 2 (#49352)" (#58523) 2021-05-20 13:20:33 -07:00
benchmark_utils Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
bottleneck_test
cpp ENH Adds dtype to nn.functional.one_hot (#58090) 2021-05-24 13:48:25 -07:00
cpp_api_parity Remove tmp folder when run unit test (#57800) 2021-05-09 20:07:14 -07:00
cpp_extensions Fix UB in library.h (#57962) 2021-05-11 16:04:02 -07:00
custom_backend
custom_operator AutoDispatchBelowAutograd takes no arguments. (#56424) 2021-04-21 14:44:12 -07:00
distributed Add GenericShardingSpec for generic tensor sharding. (#57409) 2021-05-23 16:06:05 -07:00
distributions Use torch.allclose to check tensor equality (#58429) 2021-05-17 13:16:35 -07:00
error_messages
expect Enable backward/forward compatibility for TS runtime (#57498) 2021-05-07 15:41:45 -07:00
fx Maintain submodule references during subgraph rewriting (#55463) 2021-04-30 16:46:44 -07:00
jit [JIT] integer value refinement (#56438) 2021-05-21 08:51:01 -07:00
jit_hooks
mobile [Pytorch Edge] Model Ops compatibility api (#57501) 2021-05-24 12:00:06 -07:00
onnx ENH Adds dtype to nn.functional.one_hot (#58090) 2021-05-24 13:48:25 -07:00
optim
package [torch][package] Fix importlib.resources.path for python <3.8.8 (#58718) 2021-05-21 19:16:54 -07:00
quantization [quant][fx] Validate qconfig_dict keys (#58566) 2021-05-21 15:20:05 -07:00
scripts Disallow versionless Python shebangs (#58275) 2021-05-14 08:26:02 -07:00
test_img
typing Deprecate QR (#57745) 2021-05-10 22:56:37 -07:00
HowToWriteTestsUsingFileCheck.md
linear.py
run_test.py Introduce ChunkShardingSpec as a model sharding specification. (#55728) 2021-05-23 16:04:57 -07:00
simulate_nccl_errors.py
test_ao_sparse.py [sparsity] Moving the sparsity python files to OSS (#56617) 2021-04-22 14:07:31 -07:00
test_autocast.py enable torch.cpu.amp.autocast (#57386) 2021-05-20 17:48:36 -07:00
test_autograd.py Add missing .to_sparse(ndim) gradient (#58413) 2021-05-20 15:08:34 -07:00
test_binary_ufuncs.py Fix some tensor operators to return NotImplemented for invalid inputs (#58216) 2021-05-19 13:09:57 -07:00
test_bundled_images.py [easy] Rename fb::jpeg_decode_to_NCHW to fb::image_decode_to_NCHW (#55857) 2021-04-15 13:44:13 -07:00
test_bundled_inputs.py [Pytorch] Remove run_on_bundled_input (#58344) 2021-05-17 12:44:00 -07:00
test_complex.py
test_cpp_api_parity.py
test_cpp_extensions_aot.py
test_cpp_extensions_jit.py Add minidump collection via breakpad (#55647) 2021-04-16 13:05:01 -07:00
test_cuda.py [CUDA graphs] Sync after replay (#57556) 2021-05-11 09:38:47 -07:00
test_cuda_primary_ctx.py
test_dataloader.py [DataLoader] Add generate_state for NumPy seeding (#56797) 2021-04-27 08:14:02 -07:00
test_datapipe.py [DataLoader] Fix bugs for typing (#58450) 2021-05-24 07:14:40 -07:00
test_determination.py
test_dispatch.py Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
test_expecttest.py
test_foreach.py
test_function_schema.py Add stricter check for function schemas with varargs (#56509) 2021-04-20 20:04:38 -07:00
test_functional_autograd_benchmark.py
test_futures.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_fx.py OpInfo: clone, contiguous (#58390) 2021-05-22 18:25:31 -07:00
test_fx_experimental.py OpInfo: clone, contiguous (#58390) 2021-05-22 18:25:31 -07:00
test_gen_backend_stubs.py [codegen] split out backend-specific information from NativeFunction in the model (#57361) 2021-05-17 12:25:35 -07:00
test_import_time.py Add a test to measure import torch time (#56041) 2021-04-15 00:53:30 -07:00
test_indexing.py Migrate thrust->cub for index put (#55693) 2021-04-27 12:27:09 -07:00
test_jit.py Initial Symbolic Shape Analysis (#54809) 2021-05-21 08:49:46 -07:00
test_jit_cuda_fuser.py Gelu Backward, Contribution from Kevin Stephano (#58249) 2021-05-13 19:39:39 -07:00
test_jit_disabled.py
test_jit_fuser.py
test_jit_fuser_legacy.py
test_jit_fuser_te.py Separated out working tests from not working tests for NNC OpInfo (#58788) 2021-05-22 02:24:23 -07:00
test_jit_legacy.py
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_kernel_launch_checks.py T90561249: Enforce kernel launch checks (#58178) 2021-05-13 10:41:20 -07:00
test_license.py
test_linalg.py Added sublist support for torch.einsum (#56625) 2021-05-21 08:36:45 -07:00
test_logging.py
test_metal.py
test_mkldnn.py [BE] Improve has_bf16_support (#57408) 2021-05-03 09:11:04 -07:00
test_mobile_optimizer.py s/foward/forward/g (#58497) 2021-05-19 11:42:42 -07:00
test_model_dump.py model_dump tool for model inspection (#56868) 2021-04-28 07:33:10 -07:00
test_module_init.py Support factory kwargs in torch.nn modules (#54508) 2021-04-22 16:16:53 -07:00
test_multiprocessing.py Fix mp serialization for integer nn.Parameter on CUDA (#56529) 2021-04-22 16:21:04 -07:00
test_multiprocessing_spawn.py
test_namedtensor.py
test_namedtuple_return_api.py Add torch.linalg.inv_ex without checking for errors by default (#58039) 2021-05-13 09:42:15 -07:00
test_native_functions.py
test_nn.py ENH Adds dtype to nn.functional.one_hot (#58090) 2021-05-24 13:48:25 -07:00
test_nnapi.py NNAPI: flex size support for upsample_nearest2d op (#57563) 2021-05-05 13:54:43 -07:00
test_numba_integration.py
test_numpy_interop.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_op_aliases.py [OpInfo] Add opinfo for transpose and its aliases (#56122) 2021-04-25 21:58:16 -07:00
test_openmp.py
test_ops.py Reland OpInfo support for forward AD (#58304) 2021-05-17 12:33:27 -07:00
test_optim.py
test_overrides.py Change native functions to take c10::string_view args instead of std::string (#57680) 2021-05-20 18:15:45 -07:00
test_package.py
test_profiler.py Update Kineto submodule (#58179) 2021-05-13 04:03:04 -07:00
test_pruning_op.py
test_public_bindings.py
test_pytree.py [FX] Adds PyTree support to FX through concrete_args (#55888) 2021-05-07 04:48:35 -07:00
test_quantization.py Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
test_reductions.py std/var: Return real results for complex input (#58066) 2021-05-12 03:26:55 -07:00
test_segment_reductions.py [torch][segment_reduce] Add support for initial value (#56923) 2021-04-30 18:01:31 -07:00
test_serialization.py
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py Fix tensor device in test_kthvalue_overlap (#56869) 2021-05-01 12:01:09 -07:00
test_show_pickle.py
test_sort_and_select.py Fix incorrect inplace sort in topk (#58314) (#58318) 2021-05-14 17:15:24 -07:00
test_sparse.py New: sparse complex: add linear algebra, addmm (#57129) 2021-05-07 05:37:48 -07:00
test_sparse_csr.py fixing csr addmm bug (#58768) 2021-05-24 09:54:07 -07:00
test_spectral_ops.py Remove distutils (#57040) 2021-04-29 12:10:11 -07:00
test_static_runtime.py
test_tensor_creation_ops.py Make new_ones an operator (#58405) 2021-05-17 19:24:34 -07:00
test_tensorboard.py Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
test_tensorexpr.py [TensorExpr] Re-enable skipped tests, they seem to be working now. (#58206) 2021-05-13 09:18:09 -07:00
test_tensorexpr_pybind.py Fix lint in test_tensorexpr_pybind (#57869) 2021-05-07 15:58:21 -07:00
test_testing.py Disallow versionless Python shebangs (#58275) 2021-05-14 08:26:02 -07:00
test_throughput_benchmark.py
test_torch.py add deterministic path for scatter_add_cuda for 1D tensors (#58761) 2021-05-23 21:36:02 -07:00
test_type_hints.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_type_info.py
test_type_promotion.py New: Initial support for sparse complex tensors constructors for CPU/CUDA (#57125) 2021-05-07 05:36:41 -07:00
test_typing.py Port NumPy typing testing style to PyTorch (#54234) 2021-04-15 01:25:16 -07:00
test_unary_ufuncs.py Enable geometric ops, exp2, expm1, rsqrt & erfc for BFloat16 on CUDA (#57913) 2021-05-11 09:43:05 -07:00
test_utils.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_view_ops.py Adding hsplit,vsplit and dsplit methods (#53536) 2021-04-26 09:39:09 -07:00
test_vmap.py
test_vulkan.py
test_xnnpack_integration.py