[ROCm] std::clamp work-around for hip-clang compiler

Fixes pytorch#127666.

Other std math functions are replaced with those in the global namespace
during hipify. HIP does not claim to support every function in the C++
standard library. std::clamp is not yet supported and we have been
relying on the std implementation. For Fedora 40 + gcc 14, a host-side
assert is used which is not supported. Work-around this by replacing
std::clamp with min and max for USE_ROCM builds.

Patch comes from @lamikr. Modified to use #ifndef USE_ROCM.

lamikr/rocm_sdk_builder#37

Author: jeffdaily

aten/src/ATen/native/cuda/IndexKernel.cu

@@ -259,7 +259,14 @@ void index_put_kernel_quantized_cuda(TensorIterator& iter, const IntArrayRef ind
 
     gpu_index_kernel(iter, index_size, index_stride, [inv_scale, zero_point, qmin, qmax]C10_DEVICE(char* const out_data, const char* const in_data, const int64_t offset) {
       int64_t qvalue = static_cast<int64_t>(zero_point + nearbyintf(*(float*)in_data * inv_scale));
+      // See https://github.com/pytorch/pytorch/issues/127666
+      // hip-clang std::clamp __glibcxx_assert_fail host function when building on Fedora40/gcc14
+#ifndef USE_ROCM
       qvalue = std::clamp(qvalue, qmin, qmax);
+#else
+      int64_t new_max = std::max<int64_t>(qmin, qvalue);
+      qvalue = std::min<int64_t>(qmax, new_max);
+#endif
       *(scalar_t*)(out_data + offset) = static_cast<scalar_t>(qvalue);
     });
   });