fix linking

InternLM · Oct 20, 2023 · 0fe3ab9 · 0fe3ab9
1 parent 90f5b8f
commit 0fe3ab9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -343,6 +343,7 @@ add_library(transformer-shared SHARED
   $<TARGET_OBJECTS:cuda_utils>
   $<TARGET_OBJECTS:custom_ar_comm>
   $<TARGET_OBJECTS:custom_ar_kernels>
+  $<TARGET_OBJECTS:decoder_multihead_attention>
   $<TARGET_OBJECTS:decoder_masked_multihead_attention>
   $<TARGET_OBJECTS:decoding_kernels>
   $<TARGET_OBJECTS:gpt_kernels>