input1 must be CUDA tensor

[aradhyamathur]

Hi, while passing a random tensor using the following code through the quadtree , I received the error shown below
Code:

arr = torch.randn(3, 32*32, 256)
block  = Block(256, 8, sr_ratio=2, topks=[32, 32, 32, 32])
out = block(arr, 32, 32)

where the block is a detection quadtree block , can you please guide ? Thanks.
Error :

Cell In[196], [line 113](vscode-notebook-cell:?execution_count=196&line=113)
    [110](vscode-notebook-cell:?execution_count=196&line=110)         else:
    [111](vscode-notebook-cell:?execution_count=196&line=111)             v = F.avg_pool2d(v, kernel_size=2, stride=2)
--> [113](vscode-notebook-cell:?execution_count=196&line=113) msg = self.py_att(queries, keys, values).contiguous().view(B, -1, C)
    [115](vscode-notebook-cell:?execution_count=196&line=115) x = self.proj(msg)
    [116](vscode-notebook-cell:?execution_count=196&line=116) x = self.proj_drop(x)


RuntimeError: input1 must be a CUDA tensor

This error occurs when sr_ratio > 1 does it not handle the list internally ?