hello.utils.cuda module

hello.utils.cuda.empty_cache()

hello.utils.cuda.matmul(size=1024, device='cuda:0', times=500)

hello.utils.cuda.timeit_copy(size=1024, device='cuda:0', times=500, mode='benchmark')

hello.utils.cuda.timeit_matmul(size=1024, device='cuda:0', times=500, mode='benchmark')

hello.utils.cuda.zeros(size=1024, device='cuda:0', times=500)