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
bitsandbytes
Commits
a26a321e
Commit
a26a321e
authored
Jul 10, 2023
by
Tim Dettmers
Browse files
Removed debugging statement.
parent
306f6b23
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
1 addition
and
3 deletions
+1
-3
csrc/kernels.cu
csrc/kernels.cu
+0
-2
tests/test_functional.py
tests/test_functional.py
+1
-1
No files found.
csrc/kernels.cu
View file @
a26a321e
...
@@ -3628,8 +3628,6 @@ template <typename T, int THREADS, int BITS> __global__ void kgemm_4bit_inferenc
...
@@ -3628,8 +3628,6 @@ template <typename T, int THREADS, int BITS> __global__ void kgemm_4bit_inferenc
#pragma unroll
#pragma unroll
for
(
int
k
=
0
;
k
<
num_values_4bit
;
k
++
)
for
(
int
k
=
0
;
k
<
num_values_4bit
;
k
++
)
{
{
if
((
float
)
local_A
[
k
]
<
-
10.0
f
||
(
float
)
local_B
[
k
]
<
-
10.0
f
||
local_C
>
10.0
f
)
printf
(
"%f %f = %f
\n
"
,
(
float
)
local_A
[
k
],
(
float
)
local_B
[
k
],
local_C
);
#if __CUDA_ARCH__ >= 800
#if __CUDA_ARCH__ >= 800
local_C
+=
(
float
)(
local_A
[
k
]
*
local_B
[
k
]);
local_C
+=
(
float
)(
local_A
[
k
]
*
local_B
[
k
]);
#else
#else
...
...
tests/test_functional.py
View file @
a26a321e
...
@@ -2442,7 +2442,7 @@ def test_gemv_4bit(dtype, storage_type, double_quant):
...
@@ -2442,7 +2442,7 @@ def test_gemv_4bit(dtype, storage_type, double_quant):
assert
sum
(
relerrs
)
/
len
(
relerrs
)
/
math
.
sqrt
(
dim
)
<
0.0005
assert
sum
(
relerrs
)
/
len
(
relerrs
)
/
math
.
sqrt
(
dim
)
<
0.0005
elif
dtype
==
torch
.
float32
:
elif
dtype
==
torch
.
float32
:
assert
sum
(
errs
)
/
len
(
errs
)
/
math
.
sqrt
(
dim
)
<
5e-8
assert
sum
(
errs
)
/
len
(
errs
)
/
math
.
sqrt
(
dim
)
<
5e-8
assert
sum
(
relerrs
)
/
len
(
relerrs
)
/
math
.
sqrt
(
dim
)
<
1e-
8
assert
sum
(
relerrs
)
/
len
(
relerrs
)
/
math
.
sqrt
(
dim
)
<
1e-
7
elif
dtype
==
torch
.
bfloat16
:
elif
dtype
==
torch
.
bfloat16
:
assert
sum
(
errs
)
/
len
(
errs
)
/
math
.
sqrt
(
dim
)
<
3e-4
assert
sum
(
errs
)
/
len
(
errs
)
/
math
.
sqrt
(
dim
)
<
3e-4
assert
sum
(
relerrs
)
/
len
(
relerrs
)
/
math
.
sqrt
(
dim
)
<
0.003
assert
sum
(
relerrs
)
/
len
(
relerrs
)
/
math
.
sqrt
(
dim
)
<
0.003
...
...
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