diff --git a/csrc/xpu/adam/multi_tensor_apply.dp.hpp b/csrc/xpu/adam/multi_tensor_apply.dp.hpp index 01964e1a4d38..14a130e2a23e 100644 --- a/csrc/xpu/adam/multi_tensor_apply.dp.hpp +++ b/csrc/xpu/adam/multi_tensor_apply.dp.hpp @@ -109,6 +109,12 @@ class multi_tensor_apply_kernel { std::tuple args; }; +// to make sure multi_tensor_apply_kernel can be used in sycl::buffer +namespace sycl { +template +struct is_device_copyable> : std::true_type {}; +} // namespace sycl + template void multi_tensor_apply(int block_size, int chunk_size,