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

Summary: Pull Request resolved: #1022 Reviewed By: frank-wei, muchulee8 Differential Revision: D61810808 fbshipit-source-id: 76642714422a0ee2ce2d8e224a893c98bdca2278
facebookincubator · Aug 27, 2024 · e2eff07 · e2eff07
1 parent 4360ae3
commit e2eff07
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" // @manual
+#endif
 
 #include <dlfcn.h>
 #include <torch/torch.h> // @manual=//caffe2:torch-cpp