Enable ThunderKittens in unit test

Author: xuzhao9

test/test_gpu/skip_tests_h100_pytorch.yaml

@@ -5,8 +5,6 @@
 #  op-name: to skip an entire operator
 #  op-name:\n\t- impl-name to skip an impl
 flash_attention:
-  # thunderkittens cannot handle the default input shapes
-  - tk
   # tma API changed in upstream
   - triton_tutorial_flash_v2_tma
   # triton_tutorial_*_ws kernels require triton-main

test/test_gpu/skip_tests_h100_triton_main.yaml

@@ -5,8 +5,6 @@
 #  op-name: to skip an entire operator
 #  op-name:\n\t- impl-name to skip an impl
 flash_attention:
-  # thunderkittens cannot handle the default input shapes
-  - tk
   # _ws kernels require Triton with warp specialization
   - triton_tutorial_flash_v2_ws
   - triton_tutorial_flash_v2_tma_ws