Hi,
This is expected to be slow. Though not several seconds (make sure to use torch.cuda.synchronize() properly when timing code)
You should not have to use this function for normal use. Why do you use it?