Support copying tensor from cpu to gpu without using copy engines (#10007)
This commit is contained in:
@@ -247,6 +247,7 @@ set(SOURCES
|
||||
"csrc/attention/vertical_slash_index.cu"
|
||||
"csrc/elementwise/activation.cu"
|
||||
"csrc/elementwise/cast.cu"
|
||||
"csrc/elementwise/copy.cu"
|
||||
"csrc/elementwise/fused_add_rms_norm_kernel.cu"
|
||||
"csrc/elementwise/rope.cu"
|
||||
"csrc/common_extension.cc"
|
||||
|
||||
Reference in New Issue
Block a user