AaronWang04

import os
import sys
import subprocess
import time

# Check if we're running with cuDNN enabled
cudnn_enabled = os.environ.get('TORCH_CUDNN_LAYERNORM_ENABLED') == '1'

# Import torch after environment variable is set
import torch