Indexing with slice and list

[t-vi]

Repro:

def fn(input_ids):
   return input_ids[:, [-1, 0]]

a = torch.randn(5, 5)
jfn = thunder.jit(fn)
jfn(a)

Needed for transformers BERT:

import thunder, torch, transformers

m = transformers.BertForSequenceClassification(transformers.BertConfig())
x = torch.randint(1, 10, (1, 32))
jm = thunder.jit(m)
jm(x)

Might ideally be fixed along with #460 and #187

cc @apaz-cli @tfogal

[t-vi]

@k223kim wants to work on this. Thank you!

[k223kim]

Hey Team! I will be working on this issue. Thanks!

[tfogal]

triage review:

• can we modify the checker to test for this case and not send it to nvfuser until the nvfuser bug gets fixed?
• this is related to the NeVA model
• cc @kevinstephano

[jacobhinkle]

until the nvfuser bug gets fixed?

FYI I'll merge NVIDIA/Fuser#2547 soon which should address that bug.