[Tests] Fix assert bug for fp8e5 dot on sm120.

Author: Rayfxl

python/test/unit/language/test_core.py

@@ -3480,7 +3480,7 @@ def kernel(X, stride_xm, stride_xk, Y, stride_yk, stride_yn, W, stride_wn, strid
         if capability[0] == 9 and M >= 64 and N >= 8:
             assert 'wgmma.mma_async.sync.aligned.m64n128k32.f32.e5m2.e5m2' in ptx
         elif capability[0] >= 8 and M < 64:
-            assert 'mma.sync.aligned.m16n8k16.row.col.f32.f16.f16.f32' in ptx
+            assert ('mma.sync.aligned.m16n8k16.row.col.f32.f16.f16.f32' in ptx) or ('mma.sync.aligned.m16n8k32.row.col.f32.e5m2.e5m2.f32' in ptx)
     elif in_dtype == "float8e4nv" and out_dtype == tl.float32:
         if capability[0] == 9 and M >= 64 and N >= 8:
             assert 'wgmma.mma_async.sync.aligned.m64n128k32.f32.e4m3.e4m3' in ptx