cutlass icon indicating copy to clipboard operation
cutlass copied to clipboard

[QST]How to print smem in CuteDSL

Open Dingjifeng opened this issue 1 month ago • 0 comments

tXsH0 = thr_copy_h0_s2r.partition_S(sH0[:,:,0])
                                   ^^^^^^^^^^
 File "/mnt/shared-storage-user/anaconda3/envs/fla/lib/python3.12/site-packages/nvidia_cutlass_dsl/python_packages/cutlass/cute/tensor.py", line 559, in _check_can_load_store
   raise NotImplementedError(
NotImplementedError: load & store swizzled memory is not supported yet: tensor<ptr<f32, smem, align<1024>, S<3,4,3>> o ((32,2),(8,8),(1,2)):((1,2048),(32,256),(0,4096))>

It seems unable to print smem

Dingjifeng avatar Dec 02 '25 08:12 Dingjifeng