Skip to content

Commit 21a9832

Browse files
committed
Fix FileNotFoundError on Windows in test_export_serde
1 parent f8f0ab7 commit 21a9832

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/py/dynamo/models/test_export_serde.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,9 +43,9 @@ def forward(self, x):
4343

4444
exp_program = torchtrt.dynamo.trace(model, **compile_spec)
4545
trt_module = torchtrt.dynamo.compile(exp_program, **compile_spec)
46-
torchtrt.save(trt_module, "/tmp/trt.ep", inputs=[input])
46+
torchtrt.save(trt_module, "./trt.ep", inputs=[input])
4747
# TODO: Enable this serialization issues are fixed
48-
# deser_trt_module = torchtrt.load("/tmp/trt.ep").module()
48+
# deser_trt_module = torchtrt.load("./trt.ep").module()
4949
# Check Pyt and TRT exported program outputs
5050
cos_sim = cosine_similarity(model(input), trt_module(input)[0])
5151
assertions.assertTrue(

0 commit comments

Comments
 (0)