TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

🐛 [Bug] index_put cases missing

Open apbose opened this issue 4 months ago • 0 comments

Index_put converter has the following cases missing

  1. Non consecutive indices
param( 
                test_name="nonconsecutive_caseOne", 
                source_tensor=torch.zeros([2,4,4,2], dtype=torch.float32), 
                indices_tensor=(None, torch.tensor([0, 0, 1, 1], dtype = torch.int64), None, torch.tensor([0, 0, 1, 1], dtype=torch.int64)), 
                value_tensor=torch.tensor([2, 3, 3, 4], dtype=torch.float32)
            ),
param(
                test_name="nonconsecutive_caseTwo",
                source_tensor=torch.zeros([2,4,4], dtype=torch.float32),
                indices_tensor=(torch.tensor([0, 0, 1], dtype = torch.int64), None, torch.tensor([0, 0, 1], dtype=torch.int64)),
                value_tensor=torch.tensor([2, 3, 3, 4], dtype=torch.float32),
            ),
  1. Dynamic shape support

apbose avatar Aug 28 '25 16:08 apbose