[FutureWarning] Resolving warnings related to amp.autocast#7864

drivanov wants to merge 2 commits into

dmlc:masterdmlc/dgl:masterfrom

drivanov:autocast_cudadrivanov/dgl:autocast_cudaCopy head branch name to clipboard