cutlass
cutlass copied to clipboard
[QST]How to print smem in CuteDSL
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