PaConvert
PaConvert copied to clipboard
优化GPU单测写法
PR Docs
优化GPU单测写法
PR APIs
torch.Tensor.pin_memory
torch.cuda.Event
torch.cuda.StreamContext
torch.cuda.amp.autocast
torch.cuda.current_stream
torch.cuda.device_count
torch.cuda.empty_cache
torch.cuda.get_device_capability
torch.cuda.get_device_name
torch.cuda.get_device_properties
torch.cuda.is_current_stream_capturing
torch.cuda.mem_get_info
torch.cuda.memory_usage
torch.cuda.nvtx.range_pop
torch.cuda.nvtx.range_push
torch.cuda.synchronize
torch.nn.Module.cuda
torch.sparse.addmm
torch.tensor
Thanks for your contribution!
@Xuxuanang GPU CI的测试没有通过,需要修改