From 4c37d705202f626682a5128a2fc4f4bcfb997fdc Mon Sep 17 00:00:00 2001
From: Reza Yazdani <44502768+RezaYazdaniAminabadi@users.noreply.github.com>
Date: Fri, 30 Oct 2020 12:58:53 -0700
Subject: [PATCH] fixing the AVX_256 compatibility (#497)

---
 csrc/adam/cpu_adam.cpp | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/csrc/adam/cpu_adam.cpp b/csrc/adam/cpu_adam.cpp
index 1528b28584f6..9041c2eb6a9b 100644
--- a/csrc/adam/cpu_adam.cpp
+++ b/csrc/adam/cpu_adam.cpp
@@ -384,8 +384,7 @@ void Adam_Optimizer::Step_8(float* _params,
     float w_decay = -1 * _alpha * _weight_decay;
     AVX_Data weight_decay4;
     if (_weight_decay > 0)
-        weight_decay4.data =
-            (_adamw_mode ? _mm512_set1_ps(w_decay) : _mm512_set1_ps(_weight_decay));
+        weight_decay4.data = (_adamw_mode ? SIMD_SET(w_decay) : SIMD_SET(_weight_decay));
     rounded_size = ROUND_DOWN(_param_size, (SIMD_WIDTH << 3));
 
     for (size_t t = 0; t < rounded_size; t += TILE) {