fix includes in decoder and splitk decoder (#1011)

facebookresearch · Mar 29, 2024 · 7fffd3d · 7fffd3d
1 parent 792447f
commit 7fffd3d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/xformers/csrc/attention/hip_fmha/ck_attention_forward_decoder.h b/xformers/csrc/attention/hip_fmha/ck_attention_forward_decoder.h
@@ -6,7 +6,7 @@
  */
 #pragma once
 
-#include <ck/host_utility/kernel_launch_hip.hpp>
+#include <ck/host_utility/kernel_launch.hpp>
 #include <ck/stream_config.hpp>
 #include <ck/tensor_operation/gpu/device/device_base.hpp>
 #include <ck/utility/data_type.hpp>

diff --git a/xformers/csrc/attention/hip_fmha/ck_attention_forward_decoder_splitk.h b/xformers/csrc/attention/hip_fmha/ck_attention_forward_decoder_splitk.h
@@ -1,6 +1,6 @@
 #pragma once
 
-#include <ck/host_utility/kernel_launch_hip.hpp>
+#include <ck/host_utility/kernel_launch.hpp>
 #include <ck/stream_config.hpp>
 #include <ck/tensor_operation/gpu/device/device_base.hpp>
 #include <ck/utility/data_type.hpp>