From f69ee535032eb08f00c9088dd110e86f6ab1e41e Mon Sep 17 00:00:00 2001 From: rusty1s Date: Mon, 11 Nov 2024 19:12:53 +0000 Subject: [PATCH] update --- pyg_lib/csrc/random/cpu/rand_engine.h | 25 +++++++++++++------------ 1 file changed, 13 insertions(+), 12 deletions(-) diff --git a/pyg_lib/csrc/random/cpu/rand_engine.h b/pyg_lib/csrc/random/cpu/rand_engine.h index 61aa977ca..290b940aa 100644 --- a/pyg_lib/csrc/random/cpu/rand_engine.h +++ b/pyg_lib/csrc/random/cpu/rand_engine.h @@ -134,19 +134,20 @@ class RandintEngine { for (auto& val : dst) val = static_cast((*this)(beg, end)); }; -#if WITH_MKL_BLAS() - const bool use_fallback_func = count > std::numeric_limits::max(); - if (use_fallback_func) { - fallback_func(beg, end, result); - } else { - const auto b = static_cast(beg); - const auto e = static_cast(end); - const auto c = static_cast(count); - viRngUniform(VSL_RNG_METHOD_UNIFORM_STD, stream_, c, result.data(), b, e); - } -#else + // #if WITH_MKL_BLAS() + // const bool use_fallback_func = count > + // std::numeric_limits::max(); if (use_fallback_func) { + // fallback_func(beg, end, result); + // } else { + // const auto b = static_cast(beg); + // const auto e = static_cast(end); + // const auto c = static_cast(count); + // viRngUniform(VSL_RNG_METHOD_UNIFORM_STD, stream_, c, result.data(), + // b, e); + // } + // #else fallback_func(beg, end, result); -#endif + // #endif return result; }