class_docstr: Async DMA pipeline test class.
class_docstr_tok: ["Async","DMA","pipeline","test","class","."]
docstr: Generate expected output.
docstr_tok: ["Generate","expected","output","."]
code: def expected_output(self, size_a, size_w, input_a, input_w):
      """Generate expected output."""
      if tvm.testing.utils.IS_IN_CI and (size_a > 1024 or size_w > 1):
          pytest.skip("Skipping test since it takes too long in CI.")
      expected_result = np.zeros(
          (size_a, VRMPY_SIZE_INT32), dtype="int32"
      )
      for n in range(size_a):
          for x in range(size_w):
              for index_0 in range(VRMPY_SIZE_INT32):
                  for r_index in range(4):
                      expected_result[n, index_0] += np.uint32(
                          input_a[n, index_0 * 4 + r_index]
                      ) * np.uint32(input_w[x, index_0 * 4 + r_index])
      return expected_result
  
file_path: tvm/tests/python/contrib/test_hexagon/test_async_dma_pipeline.py
imports: import numpy
  import pytest
  import tvm
  from tvm.script import tir
label_desc: Use image node_id 3 for calling the TestAsyncDMAPipeline obj's underlying member method code with example usage: obj.expected_output(size_a, size_w, input_a, input_w) and returns: expected_result