pytorch/c10
Simon Mahns 6939a56e13 [autocast][pytorch] Support autocast for MTIA (#145627)
Summary: Add autocast support to MTIA

Reviewed By: egienvalue

Differential Revision: D68572548

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145627
Approved by: https://github.com/egienvalue
2025-01-25 03:24:59 +00:00
..
benchmark Set RUNPATH so installed tests can find the required shared libraries (#136627) 2024-10-25 09:38:08 +00:00
core [autocast][pytorch] Support autocast for MTIA (#145627) 2025-01-25 03:24:59 +00:00
cuda Revert "Increase C10_COMPILE_TIME_MAX_GPUS to 128 (#144138)" 2025-01-14 19:04:12 +00:00
hip Fix hardcoded ROCm paths in Caffe2Targets.cmake (#136283) 2024-09-26 00:34:43 +00:00
macros [ROCm][Windows] Fix export macros (#144098) 2025-01-04 17:12:46 +00:00
metal [mps/inductor] Add support for erfinv. (#145643) 2025-01-24 22:55:44 +00:00
mobile [2/N] Fix extra warnings brought by clang-tidy-17 (#137459) 2024-10-08 19:05:02 +00:00
test c10::string_view -> std::string_view in pytorch (#143591) 2025-01-13 21:44:05 +00:00
util Fix PythonMod printing for C++ (#143385) 2025-01-22 14:58:35 +00:00
xpu Add low priority XPU Stream (#141119) 2024-12-31 11:15:45 +00:00
BUCK.oss
BUILD.bazel
build.bzl
CMakeLists.txt [pytorch][monitoring] Dynamic backend for WaitCounter (#135967) 2024-09-15 18:07:49 +00:00
ovrsource_defs.bzl [NeuralNetInference] Bring up iOS builds (#131917) 2024-07-30 23:01:09 +00:00