from .._ext import ffi


def get_func(name, tensor):
    name += '_'
    name += 'cuda_' if tensor.is_cuda else ''
    name += tensor.type().split('.')[-1][:-6]
    return getattr(ffi, name)