Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

covert yolo model to buddymlir failed #450

Open
georgelin0825 opened this issue Jan 8, 2025 · 0 comments
Open

covert yolo model to buddymlir failed #450

georgelin0825 opened this issue Jan 8, 2025 · 0 comments

Comments

@georgelin0825
Copy link

Describe the bug
I try to build a yolo model like example in examples resnet18 or lenet etc. when at the dynamo_compiler convert time there appear node name s1 and s2 which can't be convert make the build failed

To Reproduce

model = torch.hub.load("ultralytics/yolov5", "yolov5s", pretrained=True, trust_repo=True, skip_validation=True,autoshape=True) 
model = model.model.eval()
dynamo_compiler = DynamoCompiler(
    primary_registry=tosa.ops_registry,
    aot_autograd_decomposition=inductor_decomp,
)
data = torch.randn([1, 3, 640, 640])
# Import the model into MLIR module and parameters.
with torch.no_grad():
    graphs = dynamo_compiler.importer(model, data)

Screenshots
when I try to debug into the frontend.py It shows
Clipboard_Screenshot_1736322252

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant