Skip to content

Commit a537d9f

Browse files
committed
Uncommented some code
1 parent 7259443 commit a537d9f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

py/torch_tensorrt/dynamo/runtime/_PythonTorchTensorRTModule.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -394,9 +394,9 @@ def setup_input_tensors(
394394
+ contiguous_inputs[i + 1 :]
395395
)
396396

397-
# assert (
398-
# contiguous_inputs[i].dtype == self.input_dtypes[i]
399-
# ), f"Dtype mismatch for {i}th input({input_name}). Expect {self.input_dtypes[i]}, got {contiguous_inputs[i].dtype}."
397+
assert (
398+
contiguous_inputs[i].dtype == self.input_dtypes[i]
399+
), f"Dtype mismatch for {i}th input({input_name}). Expect {self.input_dtypes[i]}, got {contiguous_inputs[i].dtype}."
400400

401401
if need_cudagraphs_record:
402402
# If cudagraphs is enabled, this memory is reserved for future cudagraph runs

0 commit comments

Comments
 (0)