torch.arange free var issue

[anadeem2]

I noticed while compiling model that I was getting RAF free var. Using the IR_DUMP Python frames, we were able to trace the free var to be caused be torch.arange op.
In my bert_layer.py when I create position_ids for embedding like this:

position_ids = torch.arange(
            seq_length, dtype=torch.long, device=
        input_ids.device)  

I get free var issue. However, this is simply resolved by doing this instead:

        position_ids = torch.arange(
            seq_length, dtype=torch.long
        ).to(input_ids.device)

I think the solution to this might be to implemented arange op & provide lowering to raf.arange in ratex. But I did not confirm as arange op does not show in LTC IR. Regardless, in case anyone sees similar issue, this would be a good reference.