commit f27be2f781e9f247db0813d4ba3eaf76583d6878
author Jongsoo Park <jongsoo@fb.com> Tue Sep 08 18:52:31 2020 -0700
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Tue Sep 08 18:54:24 2020 -0700
tree 76b0816825c32250ee1071dc8866035b1d70809c
parent f9146b45982520d8869c7c25724e6507473ac8c7

[caffe2] fix wrong comment (#42735)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/42735

We use reduced precision only for embedding table (not for momentum) in RowWiseSparseAdagrad

Test Plan: .

Reviewed By: jianyuh

Differential Revision: D23003939

fbshipit-source-id: 062290d94b160100bc4c2f48b797833819f8e88a

caffe2/sgd/rowwise_adagrad_fused.h

diff --git a/caffe2/sgd/rowwise_adagrad_fused.h b/caffe2/sgd/rowwise_adagrad_fused.h
index ea4babb..5430b97 100644
--- a/caffe2/sgd/rowwise_adagrad_fused.h
+++ b/caffe2/sgd/rowwise_adagrad_fused.h
@@ -126,7 +126,7 @@
  * and evaluation results.
  */
 template <
-    typename Tdata, // embedding and momentum types
+    typename Tdata, // embedding types
     typename T, // everything else
     typename TLengths,
     typename rowWiseAdagradT,
@@ -390,7 +390,7 @@
 };
 
 template <
-    typename Tdata, // embedding and momentum types
+    typename Tdata, // embedding types
     typename T, // everything else
     typename TLengths,
     typename rowWiseAdagradT>
@@ -694,7 +694,7 @@
 };
 
 template <
-    typename Tdata, // embedding and momentum types
+    typename Tdata, // embedding types
     typename T, // everything else
     typename TLengths,
     typename rowWiseAdagradT>