INNER CODE UNIT · Python

sources

microsoft/muzic · museformer/museformer/kernels/range_fill/main.py:21

    sources = list(cpps) + list(cudas)

    from torch.utils.cpp_extension import load
    module = load(name='range_fill_cuda',
                  sources=sources,
                  extra_cflags=['-O2'],
                  with_cuda=True,
                  verbose=False)

    range_fill_cuda_module = module

    return range_fill_cuda_module


def range_fill(
    ranges: torch.Tensor,
    values: torch.Tensor,
    seq_len: int,

View source record →