sankeerth43 comments

Repositories
Issues
Comments

Results 1 comments of


                                            sankeerth43

RuntimeError: Unknown layout

``` import os from typing import Optional import torch from torch.utils.cpp_extension import load CUDA_FOLDER = "/home/ubuntu/anaconda3/envs/aqlm/lib/python3.10/site-packages/aqlm/inference_kernels/" CUDA_KERNEL = load( name="codebook_cuda", sources=[os.path.join(CUDA_FOLDER, "cuda_kernel.cpp"), os.path.join(CUDA_FOLDER, "cuda_kernel.cu")], ) torch.library.define( "aqlm::code1x16_matmat",...