Skip to content

Commit 5e7f1b4

Browse files
committed
Update base for Update on "introduce cuda stream into runtime backend"
This diff introduces CUDA streams into the Executorch runtime backend. The changes include: * Adding CUDA stream support to the `cuda_backend.cpp` file * Including the `cuda_runtime.h` header file in `cuda_backend.cpp` * Adding a `void* cuda_stream` field to the `AOTInductorModelContainer` struct in `aoti_model_container.h` to store the CUDA stream * Defining a new macro `ET_CHECK_OR_LOG` in `log.h` to check a condition and log an error message if the condition is false. Differential Revision: [D84128173](https://our.internmc.facebook.com/intern/diff/D84128173/) [ghstack-poisoned]
1 parent 31ea976 commit 5e7f1b4

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

backends/aoti/CMakeLists.txt

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,10 +41,7 @@ target_compile_options(aoti_common PUBLIC -fexceptions -frtti -fPIC)
4141
target_link_options(aoti_common PUBLIC -Wl,--export-dynamic)
4242

4343
# Link against PyTorch libraries and standard libraries
44-
target_link_libraries(
45-
aoti_common
46-
PUBLIC extension_tensor ${CMAKE_DL_LIBS}
47-
)
44+
target_link_libraries(aoti_common PUBLIC extension_tensor ${CMAKE_DL_LIBS})
4845
executorch_target_link_options_shared_lib(aoti_common)
4946

5047
install(

0 commit comments

Comments
 (0)