Skip to content

Commit 31ea976

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 7154361 commit 31ea976

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

backends/aoti/CMakeLists.txt

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,6 @@ target_link_options(aoti_common PUBLIC -Wl,--export-dynamic)
4444
target_link_libraries(
4545
aoti_common
4646
PUBLIC extension_tensor ${CMAKE_DL_LIBS}
47-
# Link PyTorch libraries for AOTI functions
48-
${TORCH_LIBRARIES}
4947
)
5048
executorch_target_link_options_shared_lib(aoti_common)
5149

0 commit comments

Comments
 (0)