Update fusion_record.h (#2536)

csarofeen · Mar 7, 2023 · 14930f4 · 14930f4
1 parent 12c1765
commit 14930f4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/third_party/nvfuser/csrc/python_frontend/fusion_record.h b/third_party/nvfuser/csrc/python_frontend/fusion_record.h
@@ -1974,8 +1974,8 @@ struct RandomOpRecord : RecordFunctor {
   }
 
   void operator()(FusionState& fd) final {
-    auto arg1 = fd.getFusionState(args_.at(0).index)->template as<Val>();
-    auto arg2 = fd.getFusionState(args_.at(1).index)->template as<Val>();
+    auto arg1 = fd.getFusionState(args_.at(0).index);
+    auto arg2 = fd.getFusionState(args_.at(1).index);
 
     std::vector<Val*> output_shape(output_shape_.size(), nullptr);
     std::transform(