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
..
cuda	[ROCm] fix JIT codegen (#57400)	2021-05-27 11:45:07 -07:00
fuser	[ROCm] fix JIT codegen (#57400)	2021-05-27 11:45:07 -07:00