Skip to content

Commit

Permalink
Merge branch 'backward' of github.com:Vahe1994/AQLM into backward
Browse files Browse the repository at this point in the history
  • Loading branch information
BlackSamorez committed Feb 20, 2024
2 parents 1b01426 + 5752ee8 commit abbb0d9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion inference_lib/src/aqlm/inference_kernels/triton_kernel.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ def aqlm_gemm_stupid(
raise NotImplementedError(f"Can't do Triton AQLM matmul with scales of shape {scales.shape}")

if not input.is_contiguous():
raise Exception("AAAA")
raise ValueError("Input tensor must be contiguous")

output = torch.empty(input.shape[0], out_features, device=device, dtype=dtype)
for i in range(input.shape[0]):
Expand Down

0 comments on commit abbb0d9

Please sign in to comment.