Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
text-generation-inference
Commits
083c2de9
Commit
083c2de9
authored
Dec 14, 2023
by
OlivierDehaene
Browse files
fix: fix quant linear autotune
parent
773aabdd
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
server/text_generation_server/utils/gptq/custom_autotune.py
server/text_generation_server/utils/gptq/custom_autotune.py
+1
-1
No files found.
server/text_generation_server/utils/gptq/custom_autotune.py
View file @
083c2de9
...
@@ -88,7 +88,7 @@ class Autotuner(triton.KernelInterface):
...
@@ -88,7 +88,7 @@ class Autotuner(triton.KernelInterface):
# In testings using only 40 reps seems to be close enough and it appears to be what PyTorch uses
# In testings using only 40 reps seems to be close enough and it appears to be what PyTorch uses
# PyTorch also sets fast_flush to True, but I didn't see any speedup so I'll leave the default
# PyTorch also sets fast_flush to True, but I didn't see any speedup so I'll leave the default
return
triton
.
testing
.
do_bench
(
return
triton
.
testing
.
do_bench
(
kernel_call
,
perce
ntiles
=
(
0.5
,
0.2
,
0.8
),
rep
=
40
kernel_call
,
qua
ntiles
=
(
0.5
,
0.2
,
0.8
),
rep
=
40
)
)
except
triton
.
OutOfResources
:
except
triton
.
OutOfResources
:
return
(
float
(
"inf"
),
float
(
"inf"
),
float
(
"inf"
))
return
(
float
(
"inf"
),
float
(
"inf"
),
float
(
"inf"
))
...
...
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