Restore the code used for testing.

Author: zhuyutong332

.github/workflows/pr-test-npu.yml

@@ -49,6 +49,7 @@ jobs:
               - python/**
               - test/**
               - scripts/**
+              - .github/workflows/pr-test-npu.yml
 
   test-all-build:
     needs: get-changed-files

csrc/transfer_kv_dim_exchange/op_host/transfer_kv_dim_exchange.cpp

@@ -23,8 +23,6 @@ enum TransferDirection : int64_t {
     D2H = 2,
 };
 
-// test for all
-
 // @direction: only support 1 or 2, 1 is H2D, 2 is D2H
 // @flags: only support 2
 HOST_API void transfer_kv_dim_exchange(at::Tensor &device_k, at::Tensor &host_k, at::Tensor &device_v,