We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f057b04 commit 4f5c810Copy full SHA for 4f5c810
qa/L0_infer/infer_test.py
@@ -1,6 +1,6 @@
1
#!/usr/bin/env python3
2
3
-# Copyright 2018-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright 2018-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
4
#
5
# Redistribution and use in source and binary forms, with or without
6
# modification, are permitted provided that the following conditions
@@ -264,7 +264,9 @@ def _infer_exact_helper(
264
and (output0_dtype == np.int32)
265
and (output1_dtype == np.int32)
266
):
267
- print("skipping pytorch test for int8_int32_int32")
+ print(
268
+ "skipping pytorch test for input dtype int8 and outputs dtype int32 due to a pytorch bug"
269
+ )
270
else:
271
for prefix in ensemble_prefix:
272
if "libtorch" in BACKENDS:
0 commit comments