chore: Fix missing headers in tests

peri044 · gs-olive · commit 924ebfe780dc · 2023-04-21T14:01:38.000-07:00
Signed-off-by: Dheeraj Peri &lt;peri.dheeraj@gmail.com&gt;
diff --git a/tests/core/conversion/converters/test_masked_fill.cpp b/tests/core/conversion/converters/test_masked_fill.cpp
@@ -1,3 +1,4 @@
+#include <torch/torch.h>
 #include <string>
 #include "core/compiler.h"
 #include "core/lowering/passes/passes.h"
diff --git a/tests/core/conversion/converters/test_unpack.cpp b/tests/core/conversion/converters/test_unpack.cpp
@@ -4,6 +4,8 @@
 #include "gtest/gtest.h"
 #include "tests/util/util.h"
 #include "torch/csrc/jit/ir/irparser.h"
+#include "torch/csrc/jit/passes/common_subexpression_elimination.h"
+#include "torch/torch.h"
 
 TEST(Converters, UnpackVarLowersCorrectly) {
   const auto graph = R"IR(
diff --git a/tests/core/conversion/converters/test_where.cpp b/tests/core/conversion/converters/test_where.cpp
@@ -1,3 +1,4 @@
+#include <torch/torch.h>
 #include <string>
 #include "core/compiler.h"
 #include "core/lowering/passes/passes.h"

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+#include <torch/torch.h>`
`1`	`2`	`#include <string>`
`2`	`3`	`#include "core/compiler.h"`
`3`	`4`	`#include "core/lowering/passes/passes.h"`