Skip to content

Commit b6e532b

Browse files
committed
remve unused code
1 parent 2aac0d5 commit b6e532b

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

onnxruntime/test/python/transformers/test_gqa.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1337,12 +1337,6 @@ def parity_check_gqa_past(
13371337
out_ref_np = out_ref.to(torch.float32).detach().cpu().numpy()
13381338

13391339
# --- ONNX Runtime Path ---
1340-
if ort_type == TensorProto.FLOAT16:
1341-
torch_type = torch.float16
1342-
elif ort_type == TensorProto.BFLOAT16:
1343-
torch_type = torch.bfloat16
1344-
else:
1345-
torch_type = torch.float32
13461340

13471341
q_ort, new_k_ort, new_v_ort = q, new_k, new_v
13481342
if config.packed:

0 commit comments

Comments
 (0)