comment

MPACT-ORG · Jun 26, 2024 · 4c57a61 · 4c57a61
1 parent b8164fe
commit 4c57a61
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/benchmark/python/utils/benchmark_utils.py b/benchmark/python/utils/benchmark_utils.py
@@ -136,12 +136,14 @@ def run_benchmark(
                         output.append(
                             torch.sparse_csr_tensor(*sp_out, size=dense_out.shape)
                         )
-                        # Check MPACT and torch eager both return sparse csr output.
+                        # Check MPACT and torch eager both return sparse csr output
+                        # only when torch sparse eager has been run.
                         if output_type:
                             assert output_type == torch.sparse_csr
                     else:
                         output.append(torch.from_numpy(sp_out))
-                        # Check MPACT and torch eager both return dense output.
+                        # Check MPACT and torch eager both return dense output
+                        # only when torch sparse eager has been run.
                         if output_type:
                             assert output_type == torch.strided
                     invoker, f = mpact_jit_compile(torch_net, *sparse_inputs)