profiler采集的数据还存在问题

[18liumin]

import time
use_oneflow =True

import oneflow as torch

import torch

name = "oneflow" if use_oneflow else "pytorch"
device = torch.device('cuda:0')
time_data_to_device = []

with torch.profiler.profile(
activities=[torch.profiler.ProfilerActivity.CPU, torch.profiler.ProfilerActivity.CUDA]
) as p:
for i in range(8):
x = torch.rand((1024, 32,32,16))
y = torch.rand((1024,))
start = time.perf_counter()
x = x.to(device)
y = y.to(device)
time_data_to_device.append((time.perf_counter() - start) * 1000.0)

print(name, time_data_to_device)
print(p.key_averages().table())