Fix static runtime sigrid_hash precomputed multiplier pass

Reviewed By: jfix71, pls331, houseroad Differential Revision: D54336561 fbshipit-source-id: 752deac027ef98ca429bd47611efbf65b1cbaf33
pytorch · Mar 14, 2024 · dab4b56 · dab4b56
1 parent 8866516
commit dab4b56
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/torch_glow/src/ShapeInferenceEngine.cpp b/torch_glow/src/ShapeInferenceEngine.cpp
@@ -3437,16 +3437,17 @@ ShapeInferenceEngine::argmin(const MetaStack &variableMetas) {
  *    int salt,
  *    int maxValue,
  *    Tensor multiplier_shift,
- *    bool hashIntoInt32
+ *    bool hashIntoInt32,
+ *    bool? noHashNegSalt
  * ) -> Tensor
  *
  *
  */
 Expected<TensorOutput>
 ShapeInferenceEngine::sigridHashPrecompute(const MetaStack &variableMetas) {
   RETURN_ERR_IF_NOT(
-      variableMetas.size() == 5,
-      strFormat("Expected 5 inputs, got %zu", variableMetas.size()));
+      variableMetas.size() == 6,
+      strFormat("Expected 6 inputs, got %zu", variableMetas.size()));
 
   TensorShape shape = variableMetas[0].shape<TensorShape>();