pytorch/torch/utils
Xiang Gao c88333484f [resubmit] masked_scatter thrust->cub (#58865)
Summary:
See ae7760cf50bb2cddff4663a07b9d68decf4b6c75 for the fix

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

Reviewed By: mruberry

Differential Revision: D28657940

Pulled By: ngimel

fbshipit-source-id: 9155c710b0e18ebb3bfa2dabfdd117355ac30840
2021-05-25 11:00:50 -07:00
..
backcompat
benchmark Increase mimimum number of warmup runs to 2 (#58801) 2021-05-25 08:38:52 -07:00
bottleneck
data [DataLoader] Fix bugs for typing (#58450) 2021-05-24 07:14:40 -07:00
ffi
hipify [resubmit] masked_scatter thrust->cub (#58865) 2021-05-25 11:00:50 -07:00
model_dump model_dump: Accept variable-length debug info (#57660) 2021-05-19 12:25:27 -07:00
tensorboard Remove distutils (#57040) 2021-04-29 12:10:11 -07:00
__init__.py Write mini dump on pybind exceptions (#55652) 2021-04-19 14:53:43 -07:00
_cpp_extension_versioner.py Add option for cpp_extensions to compile standalone executable (#47862) 2020-12-01 20:03:08 -08:00
_crash_handler.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
_pytree.py [FX] Adds PyTree support to FX through concrete_args (#55888) 2021-05-07 04:48:35 -07:00
bundled_inputs.py [Pytorch] Remove run_on_bundled_input (#58344) 2021-05-17 12:44:00 -07:00
checkpoint.py Support non-tensor inputs and outputs for checkpointed functions. (#52422) 2021-03-19 21:29:03 -07:00
collect_env.py Collect kernel version (#58485) 2021-05-18 10:57:59 -07:00
cpp_extension.py use importlib instead of imp as it support python 3.5+ (#57160) 2021-05-03 05:56:25 -07:00
dlpack.py
file_baton.py
hooks.py Fixes: register_full_backward_hook crash if first argument don't require a gradient (#57944) (#57945) 2021-05-11 15:07:35 -07:00
mkldnn.py BFloat16: enable prepacked weights's inference (#48922) 2021-02-17 11:20:00 -08:00
mobile_optimizer.py [Pytorch] Remove run_on_bundled_input (#58344) 2021-05-17 12:44:00 -07:00
model_zoo.py
show_pickle.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
throughput_benchmark.py