[torch-mlir] Support lowering of aten constraint ops

[praveen-g-ctt]

1. aten::sym_constrain_range
2. aten::sym_constrain_range_for_size
3. aten::_assert_scalar

[zjgarvey]

Thanks for the PR!

I'd like to request a few small changes, and I'd also like to at least have a discussion for context on these ops and their relationship with torch.symbolic_int and torch.bind_symbolic_shape.

[zjgarvey]

Thanks for the work, Praveen.

After looking at the example IR, I think it is totally fine to merge this as-is. In future patches, it would be good to add some folders (e.g., if min and max are both %none's, just erase it). And if we see other examples where these ops could inform symbolic shape constraints in a useful way for codegen, we can make changes then.

[praveen-g-ctt]

Thanks for the work, Praveen.

After looking at the example IR, I think it is totally fine to merge this as-is. In future patches, it would be good to add some folders (e.g., if min and max are both %none's, just erase it). And if we see other examples where these ops could inform symbolic shape constraints in a useful way for codegen, we can make changes then.

Sure Zach. Will add folders in future patches.

[vivekkhandelwal1]

LGTM