uzl

import torch.nn as nn
import torch.nn.functional as F
from torch.backends.cuda import sdp_kernel, SDPBackend
import time
import torch
import torch.utils.benchmark as benchmark


# check dependency
assert torch.cuda.is_available(), 'CUDA is expected.'