1. Handle None as inputs 2. Turn the function into `trace_only` 3. Fix failing CI tests Ref: - https://github.com/pytorch/pytorch/issues/131349 - https://github.com/pytorch/pytorch/issues/119363
trace_onlyRef:
aten::index_putunsupported pytorch/pytorch#119363