Method test_profile_kernel_run

test/backend/test_profiler.py:55–65

(self, wait=False)

    TestProfiler.b.uop.buffer.allocate()

  def test_profile_kernel_run(self, wait=False):
    runner_name = TestProfiler.runtime.name
    with helper_collect_profile(TestProfiler.d0) as profile:
      gs, ls = TestProfiler.prg.arg.launch_dims({})
      TestProfiler.runtime(TestProfiler.b.uop.buffer._buf, TestProfiler.a.uop.buffer._buf, global_size=gs, local_size=ls, wait=wait)

    profile, _ = helper_profile_filter_device(profile, TestProfiler.d0.device)
    kernel_runs = [x for x in profile if isinstance(x, ProfileRangeEvent)]
    assert len(kernel_runs) == 1, "one kernel run is expected"
    assert kernel_runs[0].name == runner_name, "kernel name is not correct"
    assert _dev_base(kernel_runs[0].device) == kernel_runs[0].device, "kernel should not be on a sub-device"

  def test_profile_kernel_run_wait(self):
    self.test_profile_kernel_run(wait=True)