Update EmbeddingBag.cpp

EmmettBicker · EmmettBicker · commit 7a54d87e42fd · 2024-12-10T16:36:25.000-05:00
diff --git a/aten/src/ATen/native/EmbeddingBag.cpp b/aten/src/ATen/native/EmbeddingBag.cpp
@@ -1241,9 +1241,9 @@ embedding_bag(const Tensor &weight, const Tensor &indices,
     padding_idx = maybe_wrap_dim(padding_idx, weight.size(0));
   }
   std::tuple<Tensor, Tensor, Tensor, Tensor> out;
-  bool needs_grad_path = weight.requires_grad() || 
+  bool needs_grad_path = weight.requires_grad() ||
                       weight._fw_grad(/*level=*/0).defined() ||
-                      (per_sample_weights_opt.has_value() && 
+                      (per_sample_weights_opt.has_value() &&
                        per_sample_weights_opt.value().defined() &&
                        per_sample_weights_opt.value().requires_grad());
 
