From ba37af055afc425b2fd14ee3a67c2a6c5d161c77 Mon Sep 17 00:00:00 2001
From: caozhong <zhong.z.cao@intel.com>
Date: Tue, 15 Dec 2020 14:38:19 -0800
Subject: [PATCH] Fix include files for out-of-tree compilation (#48827)

Summary:
Signed-off-by: caozhong <zhong.z.cao@intel.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/48827

Reviewed By: agolynski

Differential Revision: D25375988

Pulled By: ailzhang

fbshipit-source-id: a8d5ab4572d991d6d96dfe758011517651ff0a6b
---
 torch/csrc/jit/codegen/cuda/shape_inference.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch/csrc/jit/codegen/cuda/shape_inference.h b/torch/csrc/jit/codegen/cuda/shape_inference.h
index da2a2ed..ede73d9 100644
--- a/torch/csrc/jit/codegen/cuda/shape_inference.h
+++ b/torch/csrc/jit/codegen/cuda/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 {