Fix include files for out-of-tree compilation (#48827)

Summary: Signed-off-by: caozhong <zhong.z.cao@intel.com> Pull Request resolved: pytorch/pytorch#48827 Reviewed By: agolynski Differential Revision: D25375988 Pulled By: ailzhang fbshipit-source-id: a8d5ab4572d991d6d96dfe758011517651ff0a6b
jjsjann123 · Dec 15, 2020 · e168931 · e168931
1 parent 14a2a83
commit e168931
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/shape_inference.h b/shape_inference.h
@@ -1,6 +1,6 @@
 #pragma once
 
-#include <aten/src/ATen/Context.h>
+#include <ATen/Context.h>
 #include <torch/csrc/jit/ir/ir.h>
 
 namespace torch {