Skip to content

Commit

Permalink
Fix fd.validate function signature (#3886)
Browse files Browse the repository at this point in the history
This caused some issue in nvfuser python benchmark. Now it's fixed.
  • Loading branch information
Priya2698 authored Feb 14, 2025
1 parent 3b9ff46 commit 1386954
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nvfuser/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -559,7 +559,7 @@ def validate(
self,
inputs: list[torch.Tensor],
reference_outputs: list[torch.Tensor],
kwargs=None,
**kwargs,
):
"""
Validates the fusion outputs against the provided reference outputs, using variable tolerances determined based on datatype and reduction size.
Expand Down

0 comments on commit 1386954

Please sign in to comment.