[BUG] 使用 topk_selector 时发生非法内存访问

Traceback (most recent call last): File "/workspace_new/tilelang/examples/deepseek_v32/topk_selector.py", line 284, in test_topk_selector() File "/workspace_new/tilelang/examples/deepseek_v32/topk_selector.py", line 235, in test_topk_selector print(f"indexes shape: {indexes.shape}, indexes: {indexes}") ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/workspace_new/tilelang/venv/lib/python3.12/site-packages/torch/_tensor.py", line 1109, in format return object.format(self, format_spec) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/workspace_new/tilelang/venv/lib/python3.12/site-packages/torch/_tensor.py", line 568, in repr return torch._tensor_str._str(self, tensor_contents=tensor_contents) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/workspace_new/tilelang/venv/lib/python3.12/site-packages/torch/_tensor_str.py", line 722, in _str return _str_intern(self, tensor_contents=tensor_contents) …