v4.1 release
This commit is contained in:
@@ -120,7 +120,7 @@
|
||||
" src_vec = src.load()\n",
|
||||
" dst_vec = src_vec[indices]\n",
|
||||
" print(f\"{src_vec} -> {dst_vec}\")\n",
|
||||
" if isinstance(dst_vec, cute.TensorSSA):\n",
|
||||
" if cutlass.const_expr(isinstance(dst_vec, cute.TensorSSA)):\n",
|
||||
" dst.store(dst_vec)\n",
|
||||
" cute.print_tensor(dst)\n",
|
||||
" else:\n",
|
||||
|
||||
Reference in New Issue
Block a user