add comment

apache · Apr 12, 2022 · 78dd0e5 · 78dd0e5
1 parent a708594
commit 78dd0e5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/python/contrib/test_hexagon/benchmark_hexagon.py b/tests/python/contrib/test_hexagon/benchmark_hexagon.py
@@ -208,6 +208,10 @@ def test_one_config(dtype, sched_type, mem_scope, num_vectors_per_tensor):
         else:
             raise Exception("Unknown schedule type")
 
+        # If we're using VTCM, we *must* add a transform_layout step to the schedule.
+        # Otherwise the generated code will crash.
+        # As of 2022-04-12 the crash does not provide a useful error message to the
+        # host Python code.
         if mem_scope == "global.vtcm":
             for tensor in [A, B, C]:
                 sched[tensor].transform_layout(lambda i, j: [i, te.AXIS_SEPARATOR, j])