已合并
add test case for event elapsed_time #26732
kuhn7创建于 2025年11月19日
add test case for event elapsed_time #26732
已合并
共 1 个文件变更+12-0
| @@ -100,6 +100,18 @@ class TestAccelerator(TestCase): | |||
| 100 | ): | 100 | ): |
| 101 | event1.elapsed_time(event2) | 101 | event1.elapsed_time(event2) |
| 102 | 102 | ||
| 103 | + def test_event_elapsed_time(self): | ||
| 104 | + start_event = torch.Event(enable_timing=True) | ||
| 105 | + end_event = torch.Event(enable_timing=True) | ||
| 106 | + start_event.record() | ||
| 107 | + x = torch.randn(1000, 1000, device='npu') | ||
| 108 | + y = torch.randn(1000, 1000, device='npu') | ||
| 109 | + z = torch.matmul(x, y) | ||
| 110 | + end_event.record() | ||
| 111 | + torch.npu.synchronize() | ||
| 112 | + ms = start_event.elapsed_time(end_event) | ||
| 113 | + self.assertGreater(ms, 0) | ||
| 114 | + | ||
| 103 | 115 | ||
| 104 | if __name__ == "__main__": | 116 | if __name__ == "__main__": |
| 105 | run_tests() | 117 | run_tests() |