import torch
import torch_cuda_ext.core as core

core.print_idx()