Skip to content

Conversation

@Jagggged
Copy link

  • Add YOLOv5 test case for torch v2.8
  • Update Dockerfile to install required packages for YOLOv5
  • Add CPU fallback routine for convolution_overrideable operator

@YWHyuk YWHyuk marked this pull request as draft January 26, 2026 04:32
@YWHyuk
Copy link
Collaborator

YWHyuk commented Jan 26, 2026

Thanks for PR @Jagggged

The compilation error is caused by an excessive number of graph breaks.
Once the predefined limit is reached, the model start to be executed in eager mode.
This is why aten.convolution ~ was called.

I think increasing these limits will resolve the issue.

torch._dynamo.config.recompile_limit = 64
torch._dynamo.config.cache_size_limit = 128

Can you add the above code in the test file and remove aten.fallback code in PyTorchSimDevice?

@Jagggged Jagggged force-pushed the fix/add_yolov5_2.8_ljg branch 4 times, most recently from 0f7f1f1 to af7ebef Compare January 26, 2026 07:35
@Jagggged Jagggged force-pushed the fix/add_yolov5_2.8_ljg branch from af7ebef to 89546d7 Compare January 26, 2026 07:36
@Jagggged
Copy link
Author

As suggested, I added the two lines above, removed the fallback routine, and recommitted the changes.
Thank you for your advice!

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

Successfully merging this pull request may close these issues.

2 participants