Skip to content

Commit

Permalink
comment
Browse files Browse the repository at this point in the history
  • Loading branch information
Priya2698 committed Oct 24, 2024
1 parent 21366a1 commit 490358d
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tests/python/test_python_frontend.py
Original file line number Diff line number Diff line change
Expand Up @@ -4622,6 +4622,10 @@ def fusion_func(fd: FusionDefinition, inps) -> None:
nvf_out = fd.execute(inps, enable_options=["fuse_matmul"], disable_options=["matmul_expr_eval"], profile=True)
prof = fd.profile()
self.assertEqual(len(prof.kernel_profiles), 1)

# By default, matmul will be be run through expr_eval scheduler.
# Through setting the enable and disable options as above,
# we can execute it through matmul scheduler.
self.assertEqual(prof.kernel_profiles[0].scheduler, 'matmul')
eager_out = torch.matmul(inps[0], inps[1])
self.assertEqual(eager_out, nvf_out[0])

0 comments on commit 490358d

Please sign in to comment.