Skip to content

Commit 2b7a1a4

Browse files
shunting314meta-codesync[bot]
authored andcommitted
unittest for run2run determinism (#167482)
Summary: Not sure if the path are already properly setup so I can call 'benchmarks/dynamo/huggingface.py' in unit test directly. Let's tell from CI. X-link: pytorch/pytorch#167482 Approved by: https://github.com/v0i0, https://github.com/mlazos Reviewed By: yangw-dev Differential Revision: D87269879 fbshipit-source-id: df40728c79e62664085af32f9f5e5a121d89165c
1 parent 3078808 commit 2b7a1a4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

userbenchmark/dynamo/dynamobench/common.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2379,7 +2379,9 @@ def record_status(accuracy_status, dynamo_start_stats):
23792379
print(
23802380
f"Load model outputs from {self.args.compare_model_outputs_with} to compare"
23812381
)
2382-
saved_result = torch.load(self.args.compare_model_outputs_with)
2382+
saved_result = torch.load(
2383+
self.args.compare_model_outputs_with, weights_only=False
2384+
)
23832385
is_bitwise_same = bitwise_same(saved_result, new_result)
23842386
if not is_bitwise_same:
23852387
print(

0 commit comments

Comments
 (0)