Skip to content

Commit 1041576

Browse files
committed
Support dcu to use Event to measure performance.
1 parent bdc8d03 commit 1041576

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

graph_net/paddle/test_compiler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ def measure_performance(model_call, args, synchronizer_func, profile=False):
138138
flush=True,
139139
)
140140

141-
if "cuda" in args.device:
141+
if "cuda" in args.device or "dcu" in args.device:
142142
"""
143143
Acknowledgement: We evaluate the performance on both end-to-end and GPU-only timings,
144144
With reference to methods only based on CUDA events from KernelBench in https://github.com/ScalingIntelligence/KernelBench

0 commit comments

Comments
 (0)