bug fixes in sgd kernel in bfp16 bringup

98a64039 · lcskrishna · b2da92fc · 98a64039
Commit 98a64039 authored May 20, 2020 by lcskrishna
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

csrc/multi_tensor_sgd_kernel.cu csrc/multi_tensor_sgd_kernel.cu +1 -1

No files found.
--- a/csrc/multi_tensor_sgd_kernel.cu
+++ b/csrc/multi_tensor_sgd_kernel.cu
@@ -271,7 +271,7 @@ void multi_tensor_sgd_cuda(
        scale);
  }
  // Case 5. bfp16, bfp16, bfp16, No
-  if(grad_type == at::ScalarType::BFloat16 &&
+  else if(grad_type == at::ScalarType::BFloat16 &&
     weight_type == at::ScalarType::BFloat16 &&
     num_tensors == 3)
  {