Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
d20e2611
Unverified
Commit
d20e2611
authored
Mar 24, 2025
by
Qubitium-ModelCloud
Committed by
GitHub
Mar 24, 2025
Browse files
Fix non-contiguous input passed to Marlin kernel (#15319)
parent
f622dbcf
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
0 deletions
+4
-0
vllm/model_executor/layers/quantization/kernels/mixed_precision/marlin.py
...tor/layers/quantization/kernels/mixed_precision/marlin.py
+4
-0
No files found.
vllm/model_executor/layers/quantization/kernels/mixed_precision/marlin.py
View file @
d20e2611
...
...
@@ -115,6 +115,10 @@ class MarlinLinearKernel(MPLinearKernel):
layer
:
torch
.
nn
.
Module
,
x
:
torch
.
Tensor
,
bias
:
Optional
[
torch
.
Tensor
]
=
None
)
->
torch
.
Tensor
:
# marlin requires contiguous memory layout
# prefix caching may cause x to be non-contiguous
x
=
x
.
contiguous
()
# no-op if already contiguous
c
=
self
.
config
w_q
,
w_s
,
w_zp
,
w_gidx
=
self
.
_get_weight_params
(
layer
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment