leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Update blackwell tutorial to be compatible with 4.5-dev version (#3130)

Browse Source 
* Update blackwell tutorial to be compatible with 4.5-dev version

* update example for reverted changes

* add more example fix
This commit is contained in:
Longsheng Du
2026-04-09 14:40:33 +08:00
committed by GitHub
parent bd01dd3651
commit 08185b9c3e
12 changed files with 29 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/python/CuTeDSL/notebooks/tour_to_sol_gemm.ipynb
View File

@@ -369,7 +369,7 @@
" num_threads=threads_per_cta,\n",
" )\n",
" tmem = utils.TmemAllocator(\n",
" storage.tmem_holding_buf,\n",
" storage.tmem_holding_buf.ptr,\n",
" barrier_for_retrieve=tmem_alloc_barrier,\n",
" )\n",
" num_tmem_cols = 512\n",
@@ -742,7 +742,7 @@
" num_threads=threads_per_cta,\n",
" )\n",
" tmem = utils.TmemAllocator(\n",
" storage.tmem_holding_buf,\n",
" storage.tmem_holding_buf.ptr,\n",
" barrier_for_retrieve=tmem_alloc_barrier,\n",
" )\n",
" num_tmem_cols = 512\n",