Use the full header file path to fix the build (#1022)

Summary: Pull Request resolved: #1022 Differential Revision: D61810808
facebookincubator · Aug 26, 2024 · b7272ab · b7272ab
1 parent 4360ae3
commit b7272ab
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/fx2ait/fx2ait/csrc/AITModelImpl.h b/fx2ait/fx2ait/csrc/AITModelImpl.h
@@ -14,7 +14,11 @@
 //
 #pragma once
 
-#include "model_interface.h" // @manual=//aitemplate/AITemplate/static/include:aitemplate
+#ifdef FBCODE_AIT
+#include "aitemplate/AITemplate/static/include/model_interface.h" // @manual=//aitemplate/AITemplate/static/include:aitemplate
+#else
+#include "model_interface.h"
+#endif
 
 #include <dlfcn.h>
 #include <torch/torch.h> // @manual=//caffe2:torch-cpp