Update several paths and targets that drifted upstream. (#13)

iree-org · Mar 3, 2023 · 587caef · 587caef
1 parent 802376c
commit 587caef
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 7 deletions.
diff --git a/iree/integrations/pjrt/common/BUILD b/iree/integrations/pjrt/common/BUILD
@@ -31,7 +31,7 @@ iree_pjrt_cc_library(
         "@iree_core//runtime/src/iree/hal",
         "@iree_core//runtime/src/iree/modules/hal",
         "@iree_core//runtime/src/iree/vm",
-        "@iree_core//runtime/src/iree/vm:bytecode_module",
+        "@iree_core//runtime/src/iree/vm/bytecode:module",
         "@org_tensorflow//tensorflow/compiler/xla:shape_util",
         "@org_tensorflow//tensorflow/compiler/xla/pjrt/c:pjrt_c_api_hdrs",
     ],
@@ -46,9 +46,10 @@ iree_pjrt_cc_library(
         "dylib_platform.h",
     ],
     deps = [
-        ":impl",
         ":debugging",
-        "@iree_core//compiler/src/iree/compiler/API2/Stub",
+        ":impl",
+        "@iree_core//compiler/bindings/c:headers",
+        "@iree_core//compiler/bindings/c:loader",
         "@iree_core//runtime/src/iree/base/internal:file_io",
         "@iree_core//runtime/src/iree/base/internal:path",
     ],
@@ -64,7 +65,7 @@ iree_pjrt_cc_library(
     ],
     deps = [
         ":debugging",
-        "@iree_core//compiler/src/iree/compiler/API2:EmbedHeaders",
+        "@iree_core//compiler/bindings/c:headers",
     ],
 )
 

diff --git a/iree/integrations/pjrt/common/api_impl.h b/iree/integrations/pjrt/common/api_impl.h
@@ -19,7 +19,7 @@
 #include "iree/integrations/pjrt/common/platform.h"
 #include "iree/modules/hal/module.h"
 #include "iree/vm/api.h"
-#include "iree/vm/bytecode_module.h"
+#include "iree/vm/bytecode/module.h"
 #include "tensorflow/compiler/xla/pjrt/c/pjrt_c_api.h"
 #include "tensorflow/compiler/xla/shape_util.h"
 

diff --git a/iree/integrations/pjrt/common/compiler.h b/iree/integrations/pjrt/common/compiler.h
@@ -10,7 +10,7 @@
 #include <memory>
 #include <string>
 
-#include "iree/compiler/API2/Embed.h"
+#include "iree/compiler/embedding_api.h"
 #include "iree/integrations/pjrt/common/debugging.h"
 
 namespace iree::pjrt {

diff --git a/iree/integrations/pjrt/common/dylib_platform.cc b/iree/integrations/pjrt/common/dylib_platform.cc
@@ -11,7 +11,7 @@
 
 #include "iree/base/internal/file_io.h"
 #include "iree/base/internal/path.h"
-#include "iree/compiler/API2/Stub/Loader.h"
+#include "iree/compiler/loader.h"
 
 namespace iree::pjrt {