hacky support .so file seperation by hardcoding file path

Author: Gasoonjia

backends/aoti/aoti_backend.py

@@ -16,7 +16,8 @@
     PreprocessResult,
 )
 from executorch.exir.backend.compile_spec_schema import CompileSpec
-
+import os
+import shutil
 
 @final
 class AotiBackend(BackendDetails):
@@ -31,13 +32,18 @@ def preprocess(
         copy_edge_program = copy.deepcopy(edge_program)
         graph_module = copy_edge_program.graph_module
         args, kwargs = copy_edge_program.example_inputs
-        so_path = torch._inductor.aot_compile(graph_module, args, kwargs, options={})  # type: ignore[arg-type]
-        print(so_path)
+        temp_so_path = torch._inductor.aot_compile(graph_module, args, kwargs, options={})  # type: ignore[arg-type]
+        so_path = os.path.join(os.getcwd(), 'aoti.so')
+        print("so_path after aot_compile: ", temp_so_path)
+        print("so path we will using ", so_path)
+        shutil.copyfile(temp_so_path, so_path)
+
         check_call(
             f"patchelf --remove-needed libtorch.so --remove-needed libc10.so --remove-needed libtorch_cuda.so --remove-needed libc10_cuda.so --remove-needed libtorch_cpu.so --add-needed libcudart.so {so_path}",
             shell=True,
         )
 
-        with open(so_path, "rb") as f:
-            data = f.read()
-        return PreprocessResult(data)
+        # with open(so_path, "rb") as f:
+        #     data = f.read()
+
+        return PreprocessResult(so_path.encode("utf-8"))

backends/aoti/runtime/AotiBackend.cpp

@@ -430,20 +430,24 @@ class AOTIBackend final : public ::executorch::runtime::BackendInterface {
     // We could load the .so content directly. But I don't want to deal with
     // relocation. So dumping a file and using dlopen
 
-    // Create a temporary file
-    std::ofstream outfile("/tmp/test.so", std::ios::binary);
+    // // Create a temporary file
+    // std::ofstream outfile("/tmp/test.so", std::ios::binary);
 
-    // Write the ELF buffer to the temporary file
-    outfile.write((char*)processed->data(), sizeof(void*) * processed->size());
+    // // Write the ELF buffer to the temporary file
+    // outfile.write((char*)processed->data(), sizeof(void*) * processed->size());
 
-    // Finish writing the file to disk
-    outfile.close();
+    // // Finish writing the file to disk
+    // outfile.close();
 
-    // Free the in-memory buffer
-    processed->Free();
+    // // Free the in-memory buffer
+    // processed->Free();
+
+    const char* so_path = static_cast<const char*>(processed->data());
+
+    printf("so path: %s\n", so_path);
 
     // Load the ELF using dlopen
-    void* so_handle = dlopen("/tmp/test.so", RTLD_LAZY | RTLD_LOCAL);
+    void* so_handle = dlopen(so_path, RTLD_LAZY | RTLD_LOCAL);
     if (so_handle == nullptr) {
       std::cout << dlerror() << std::endl;
       return Error::AccessFailed;