Skip to content

Commit 442c445

Browse files
committed
install header
1 parent dcb9915 commit 442c445

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/ATen/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ install_xpu_headers("native/quantized/xpu/sycl")
4040
install_xpu_headers("native/sparse/xpu")
4141
install_xpu_headers("native/sparse/xpu/sycl")
4242
install_xpu_headers("native/transformers/xpu")
43-
install_xpu_headers("native/transformers/xpu/sycl")
43+
install_xpu_headers("native/transformers/xpu/flash_attn")
4444

4545
if(xpu_ops_generated_headers)
4646
install(FILES ${xpu_ops_generated_headers} DESTINATION ${AT_INSTALL_INCLUDE_DIR}/ATen/ops)

0 commit comments

Comments
 (0)