cuda : disables launch_fattn PDL enrollment due to compiler bug (#23825)
This commit is contained in:
@@ -1153,8 +1153,8 @@ void launch_fattn(
|
|||||||
|
|
||||||
GGML_ASSERT(block_dim.x % warp_size == 0);
|
GGML_ASSERT(block_dim.x % warp_size == 0);
|
||||||
|
|
||||||
const ggml_cuda_kernel_launch_params launch_params = ggml_cuda_kernel_launch_params(blocks_num, block_dim, nbytes_shared, main_stream);
|
// disabled PDL enrollment for now due to a compiler bug.
|
||||||
ggml_cuda_kernel_launch(fattn_kernel, launch_params,
|
fattn_kernel<<<blocks_num, block_dim, nbytes_shared, main_stream>>>(
|
||||||
(const char *) Q->data,
|
(const char *) Q->data,
|
||||||
K_data,
|
K_data,
|
||||||
V_data,
|
V_data,
|
||||||
|
|||||||
Reference in New Issue
Block a user