From ee523ad9e5aae20b4bdc60cb53fed578de3d07d3 Mon Sep 17 00:00:00 2001 From: Yingge He Date: Tue, 16 Jun 2026 18:04:35 -0700 Subject: [PATCH] test: Simplify BF16 triton dtype selection in identity_test Signed-off-by: Yingge He --- qa/L0_backend_identity/identity_test.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/qa/L0_backend_identity/identity_test.py b/qa/L0_backend_identity/identity_test.py index b96649241d..7f1cc6fb70 100755 --- a/qa/L0_backend_identity/identity_test.py +++ b/qa/L0_backend_identity/identity_test.py @@ -212,10 +212,7 @@ in0 = 16384 * np.ones(shape, dtype="int") in0n = np.array([str(x) for x in in0.reshape(in0.size)], dtype=object) input_data = in0n.reshape(in0.shape) - if model_name != "identity_bf16": - triton_type = np_to_triton_dtype(input_data.dtype) - else: - triton_type = "BF16" + triton_type = np_to_triton_dtype(input_data.dtype) inputs = [client_util.InferInput("INPUT0", input_data.shape, triton_type)] inputs[0].set_data_from_numpy(input_data)