pytorch/cmake
Jeff Daily ba694520e5 [ROCm] fix JIT codegen (#57400)
Summary:
Fixes upcoming changes that are part of ROCm 4.2 and affect PyTorch JIT.

- ROCM_VERSION macro must be available to both device and host compilation passes.
- Unifies some of CUDA and HIP differences in the code generated.
  - NAN / POS_INFINITY / NEG_INFINITY
  - Do not hipify `extern __shared__` -> `HIP_DYNAMIC_SHARED()` macro [deprecated]
- Differentiates bf16 codegen for HIP.
- Optionally provides missing macros when using hiprtc precompiled header feature.

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

Reviewed By: ejguan

Differential Revision: D28421065

Pulled By: malfet

fbshipit-source-id: 215f476773c61d8b0d9d148a4e5f5d016f863074
2021-05-27 11:45:07 -07:00
..
External cmake: Enable (s)ccache for nccl builds (#55814) 2021-04-13 14:49:25 -07:00
Modules Adding a new include directory in BLIS search path (#58166) 2021-05-24 08:57:02 -07:00
Modules_CUDA_fix Lint trailing newlines (#54737) 2021-03-30 13:09:52 -07:00
public Move cublas dependency after CuDNN (#58287) 2021-05-24 09:39:09 -07:00
Allowlist.cmake Replace whitelist with allowlist (#42067) 2020-07-28 08:01:16 -07:00
BuildVariables.cmake Remove Caffe2_MAIN_LIBS (#38408) 2020-05-15 12:27:15 -07:00
Caffe2Config.cmake.in Formatting cmake (to lowercase without space for if/elseif/else/endif) (#35521) 2020-03-27 14:25:17 -07:00
Caffe2ConfigVersion.cmake.in Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
cmake_uninstall.cmake.in Formatting cmake (to lowercase without space for if/elseif/else/endif) (#35521) 2020-03-27 14:25:17 -07:00
Codegen.cmake [pytorch] reintroduce static dispatch (#51957) 2021-02-19 11:41:39 -08:00
DebugHelper.cmake Add debug helper function to check target property (#52093) 2021-02-11 15:37:14 -08:00
Dependencies.cmake [ROCm] fix JIT codegen (#57400) 2021-05-27 11:45:07 -07:00
GoogleTestPatch.cmake Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
iOS.cmake Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
Metal.cmake [GPU] Enable Metal on macosx (#47635) 2020-11-17 14:44:34 -08:00
MiscCheck.cmake Fix warning when running scripts/build_ios.sh (#49457) 2021-01-11 19:31:32 -08:00
ProtoBuf.cmake Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
ProtoBufPatch.cmake Patch generate files for system protobuf (#44583) 2020-09-29 18:06:33 -07:00
Summary.cmake Move cublas dependency after CuDNN (#58287) 2021-05-24 09:39:09 -07:00
TorchConfig.cmake.in Expand Kineto platform support (ci-all) (#56323) 2021-05-05 15:00:01 -07:00
TorchConfigVersion.cmake.in Formatting cmake (to lowercase without space for if/elseif/else/endif) (#35521) 2020-03-27 14:25:17 -07:00
VulkanCodegen.cmake [vulkan] Fix glslc path for desktop build (#56507) 2021-05-11 14:18:39 -07:00
VulkanDependencies.cmake [Vulkan]Fix build warnings-treated-as-error on Linux. (#52781) 2021-03-03 13:48:43 -08:00