Caught an unknown exception!

[PANFEI-CHENG]

当我令stratified_transformer.channels = [256]，stratified_transformer.num_heads = [4]时，我遇到了Caught an unknown exception!这个错误。我发现在pontops2中仅提供了
switch (C / h) {
case 16:
attention_step1_forward_cuda_kernel_v2<16><<<blocks, n_threads, 0>>>(N, M, h, q, k, index0_offsets, index1, attn);
break;
case 32:
attention_step1_forward_cuda_kernel_v2<32><<<blocks, n_threads, 0>>>(N, M, h, q, k, index0_offsets, index1, attn);
break;
default:
throw "d != 16 and d != 32";
}
我能否简单的修改这里的16为256/4=64来解决该bug？