[Bug] cu128 test_bf16 error

[FlamingoPg]

I build with B200 cu128, test_bf16 failed:

➜  DeepGEMM git:(main) python3 /sgl-workspace/DeepGEMM/tests/test_bf16.py                                           
Library path:
 > ['/usr/local/lib/python3.12/dist-packages/deep_gemm']

Testing GEMM:
Warning: please use at least NVCC 12.9 for the best DeepGEMM performanceNVCC compilation failed: nvcc fatal   : Unsupported gpu architecture 'sm_100f'

Traceback (most recent call last):
  File "/sgl-workspace/DeepGEMM/tests/test_bf16.py", line 123, in <module>
    test_gemm()
  File "/sgl-workspace/DeepGEMM/tests/test_bf16.py", line 30, in test_gemm
    getattr(deep_gemm, func_name)(a, b, d, c=c)
RuntimeError: Assertion error (csrc/apis/../jit_kernels/impls/../../jit/compiler.hpp:176): false and "NVCC compilation failed"

[FlamingoPg]

I guess this PR break #157

[FlamingoPg]

test_fp8 failed too.

[zhyncs]

@LyricZhao May you help take a look? Thanks!

[FlamingoPg]

I tried to fix it lol, #173

[LyricZhao]

Thanks! Closing it now.