Open
[Fix] Remove hard-coded property target. #60
hanxu88Add on 8 days ago
[Fix] Remove hard-coded property target. #60
Open
Merged fromhanxu88/ops-tilelanggraphintocann/ops-tilelangmaster
Total: 1 files changed+1-1
| @@ -111,7 +111,7 @@ def mhc_sinkhorn_forward_kernel( | |||
| 111 | out[token_start + store_token, row, col] = matrix_ub[row, col, store_token] | 111 | out[token_start + store_token, row, col] = matrix_ub[row, col, store_token] |
| 112 | 112 | ||
| 113 | 113 | ||
| 114 | -@tilelang.jit(target='pto', pass_configs=_PASS_CONFIGS) | 114 | +@tilelang.jit(pass_configs=_PASS_CONFIGS) |
| 115 | def mhc_sinkhorn_backward_kernel( | 115 | def mhc_sinkhorn_backward_kernel( |
| 116 | grad_output, | 116 | grad_output, |
| 117 | x, | 117 | x, |