python3 tests/python/sgl_kernel_npu/test_transfer_kv_dim_exchange.py
kv transfer D2H, v_empty=True, index_k_empty=True, 2d copy times is 30, total copy size is 251658240 bytes, total duration 21.995ms
.kv transfer H2D, v_empty=True, index_k_empty=True, 2d copy times is 30, total copy size is 251658240 bytes, total duration 42.014ms
.kv transfer D2H, v_empty=False, index_k_empty=True, 2d copy times is 60, total copy size is 503316480 bytes, total duration 102.387ms
.kv transfer H2D, v_empty=False, index_k_empty=True, 2d copy times is 60, total copy size is 503316480 bytes, total duration 95.783ms
.kv transfer D2H, v_empty=False, index_k_empty=False, 2d copy times is 90, total copy size is 754974720 bytes, total duration 142.581ms
.kv transfer H2D, v_empty=False, index_k_empty=False, 2d copy times is 90, total copy size is 754974720 bytes, total duration 140.975ms
.
----------------------------------------------------------------------
Ran 6 tests in 7.970s

OK