Skip to content

Commit 4f5c810

Browse files
authored
test: minor improvement on L0_infer infer_test.py skip pytorch test output msg (#7971)
1 parent f057b04 commit 4f5c810

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

qa/L0_infer/infer_test.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
#!/usr/bin/env python3
22

3-
# Copyright 2018-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3+
# Copyright 2018-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
44
#
55
# Redistribution and use in source and binary forms, with or without
66
# modification, are permitted provided that the following conditions
@@ -264,7 +264,9 @@ def _infer_exact_helper(
264264
and (output0_dtype == np.int32)
265265
and (output1_dtype == np.int32)
266266
):
267-
print("skipping pytorch test for int8_int32_int32")
267+
print(
268+
"skipping pytorch test for input dtype int8 and outputs dtype int32 due to a pytorch bug"
269+
)
268270
else:
269271
for prefix in ensemble_prefix:
270272
if "libtorch" in BACKENDS:

0 commit comments

Comments
 (0)