INNER CODE UNIT · Python

sources

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

    sources = list(cpps) + list(cudas)

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

    block_fill_cuda_module = module

    return block_fill_cuda_module


def block_fill_(out, block_ranges, head_masks, fill_value, no_cuda_kernel=False):
    """

    :param block_ranges: (num_blocks, 4)

View source record →