# RUN: %PYTHON %s
import torch_mlir.ir
from torch_mlir.dialects import torch
with torch_mlir.ir.Context() as ctx:
torch.register_dialect(ctx)