Fix/isfinite on windows (#24927)

* refine isfinite, test=develop

* use namespace std of isfinite, test=develop, test=win_gpu
revert-24981-add_device_attr_for_regulization
Leo Chen 5 years ago committed by GitHub
parent a9dca5805a
commit a7cb97a1a5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -25,7 +25,7 @@ __global__ void AmpCheckFiniteAndScale(const T* in, const T* scale, int num,
const int idx = threadIdx.x + blockIdx.x * blockDim.x;
if (idx < num) {
if (!std::isfinite(in[idx])) {
if (!isfinite(in[idx])) {
*found_inf = 1;
}
out[idx] = *found_inf ? in[idx] : in[idx] * scale[0];

Loading…
Cancel
Save