import torch
import trme_compiler

def test_compiler():
    print("🔧 Testing TRME Graph Compiler...")

    model = torch.nn.Sequential(
        torch.nn.Linear(32, 32),
        torch.nn.ReLU(),
        torch.nn.Linear(32, 10)
    )

    opt_model = trme_compiler.compile_model(model)

    input = torch.randn(1, 32)
    output = opt_model(input)
    print("✅ Compiler Pass executed.")

if __name__ == "__main__":
    test_compiler()
