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
953b7285
"tests/git@developer.sourcefind.cn:OpenDAS/llama-factory.git" did not exist on "849877159ac0c6cc6397decc31e2b1073d765950"
Commit
953b7285
authored
Jul 26, 2022
by
Tim Dettmers
Browse files
Fixed cpuonly build.
parent
f2dd7032
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
146 additions
and
132 deletions
+146
-132
csrc/pythonInterface.c
csrc/pythonInterface.c
+2
-2
deploy_from_slurm.sh
deploy_from_slurm.sh
+143
-129
setup.py
setup.py
+1
-1
No files found.
csrc/pythonInterface.c
View file @
953b7285
...
...
@@ -82,7 +82,6 @@ void quantizeBlockwise_stochastic_fp32(float * code, float *A, float *absmax, un
void
dequantizeBlockwise_fp16
(
float
*
code
,
unsigned
char
*
A
,
float
*
absmax
,
half
*
out
,
int
blocksize
,
const
int
n
){
dequantizeBlockwise
<
half
>
(
code
,
A
,
absmax
,
out
,
blocksize
,
n
);
}
\
void
dequantizeBlockwise_fp32
(
float
*
code
,
unsigned
char
*
A
,
float
*
absmax
,
float
*
out
,
int
blocksize
,
const
int
n
){
dequantizeBlockwise
<
float
>
(
code
,
A
,
absmax
,
out
,
blocksize
,
n
);
}
#endif
#define MAKE_FUNC_TRANSFORM(fbits, fsrc, ftrgt, ftranspose, dtype, src, target, transpose, bits) \
void transform_##fbits##_##fsrc##_to_##ftrgt##_##ftranspose(cublasLtHandle_t ltHandle, dtype *A, dtype *out, int dim1, int dim2) \
...
...
@@ -129,10 +128,11 @@ void spmm_coo_very_sparse_naive_fp16(int *max_count, int *max_idx, int *offset_r
void
spmm_coo_very_sparse_naive_int8
(
int
*
max_count
,
int
*
max_idx
,
int
*
offset_rowidx
,
int
*
rowidx
,
int
*
colidx
,
half
*
values
,
signed
char
*
B
,
half
*
out
,
float
*
dequant_stats
,
int
nnz_rows
,
int
nnz
,
int
rowsA
,
int
rowsB
,
int
colsB
)
{
spmm_coo_very_sparse_naive
<
signed
char
,
8
>
(
max_count
,
max_idx
,
offset_rowidx
,
rowidx
,
colidx
,
values
,
B
,
out
,
dequant_stats
,
nnz_rows
,
nnz
,
rowsA
,
rowsB
,
colsB
);
}
#endif
extern
"C"
{
#if BUILD_CUDA
#if BUILD_CUDA
void
cestimate_quantiles_fp32
(
float
*
A
,
float
*
code
,
float
offset
,
int
n
){
estimateQuantiles_fp32
(
A
,
code
,
offset
,
n
);
}
void
cestimate_quantiles_fp16
(
half
*
A
,
float
*
code
,
float
offset
,
int
n
){
estimateQuantiles_fp16
(
A
,
code
,
offset
,
n
);
}
void
cquantize
(
float
*
code
,
float
*
A
,
unsigned
char
*
out
,
int
n
){
quantize
(
code
,
A
,
out
,
n
);
}
...
...
deploy_from_slurm.sh
View file @
953b7285
...
...
@@ -4,117 +4,131 @@ BASE_PATH=$1
module unload cuda
module unload gcc
#rm -rf dist build
#make clean
#make cleaneggs
#export CUDA_HOME=$BASE_PATH/cuda-11.0
#make cuda110
#
#if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# # Control will enter here if $DIRECTORY doesn't exist.
# echo "Compilation unsuccessul!" 1>&2
# exit 64
#fi
##CUDA_VERSION=110 python -m build
##python -m twine upload dist/* --verbose
#
#rm -rf dist build
#make clean
#make cleaneggs
#export CUDA_HOME=$BASE_PATH/cuda-11.1
#make cuda11x
#
#if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# # Control will enter here if $DIRECTORY doesn't exist.
# echo "Compilation unsuccessul!" 1>&2
# exit 64
#fi
##CUDA_VERSION=111 python -m build
##python -m twine upload dist/* --verbose
#
#rm -rf dist build
#make clean
#make cleaneggs
#export CUDA_HOME=$BASE_PATH/cuda-11.2
#make cuda11x
#
#if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# # Control will enter here if $DIRECTORY doesn't exist.
# echo "Compilation unsuccessul!" 1>&2
# exit 64
#fi
##CUDA_VERSION=112 python -m build
##python -m twine upload dist/* --verbose
#
#rm -rf dist build
#make clean
#make cleaneggs
#export CUDA_HOME=$BASE_PATH/cuda-11.3
#make cuda11x
#
#if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# # Control will enter here if $DIRECTORY doesn't exist.
# echo "Compilation unsuccessul!" 1>&2
# exit 64
#fi
##CUDA_VERSION=113 python -m build
##python -m twine upload dist/* --verbose
#
#rm -rf dist build
#make clean
#make cleaneggs
#export CUDA_HOME=$BASE_PATH/cuda-11.4
#make cuda11x
#
#if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# # Control will enter here if $DIRECTORY doesn't exist.
# echo "Compilation unsuccessul!" 1>&2
# exit 64
#fi
##CUDA_VERSION=114 python -m build
###python -m twine upload dist/* --verbose
#
#rm -rf dist build
#make clean
#make cleaneggs
#export CUDA_HOME=$BASE_PATH/cuda-11.5
#make cuda11x
#
#if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# # Control will enter here if $DIRECTORY doesn't exist.
# echo "Compilation unsuccessul!" 1>&2
# exit 64
#fi
##CUDA_VERSION=115 python -m build
##python -m twine upload dist/* --verbose
#
#rm -rf dist build
#make clean
#make cleaneggs
#export CUDA_HOME=$BASE_PATH/cuda-11.6
#
#make cuda11x
#if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# # Control will enter here if $DIRECTORY doesn't exist.
# echo "Compilation unsuccessul!" 1>&2
# exit 64
#fi
##CUDA_VERSION=116 python -m build
##python -m twine upload dist/* --verbose
#
#rm -rf dist build
#make clean
#make cleaneggs
#export CUDA_HOME=$BASE_PATH/cuda-11.7
#make cuda11x
#
#if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
# # Control will enter here if $DIRECTORY doesn't exist.
# echo "Compilation unsuccessul!" 1>&2
# exit 64
#fi
##CUDA_VERSION=117 python -m build
##python -m twine upload dist/* --verbose
rm
-rf
dist build
make clean
make cleaneggs
export
CUDA_HOME
=
make cpuonly
if
[
!
-f
"./bitsandbytes/libbitsandbytes.so"
]
;
then
# Control will enter here if $DIRECTORY doesn't exist.
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
CUDA_VERSION
=
cpu python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
make cleaneggs
export
CUDA_HOME
=
$BASE_PATH
/cuda-11.0
make cuda110
if
[
!
-f
"./bitsandbytes/libbitsandbytes.so"
]
;
then
# Control will enter here if $DIRECTORY doesn't exist.
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
CUDA_VERSION
=
110 python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
make cleaneggs
export
CUDA_HOME
=
$BASE_PATH
/cuda-11.1
make cuda11x
if
[
!
-f
"./bitsandbytes/libbitsandbytes.so"
]
;
then
# Control will enter here if $DIRECTORY doesn't exist.
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
CUDA_VERSION
=
111 python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
make cleaneggs
export
CUDA_HOME
=
$BASE_PATH
/cuda-11.2
make cuda11x
if
[
!
-f
"./bitsandbytes/libbitsandbytes.so"
]
;
then
# Control will enter here if $DIRECTORY doesn't exist.
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
CUDA_VERSION
=
112 python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
make cleaneggs
export
CUDA_HOME
=
$BASE_PATH
/cuda-11.3
make cuda11x
if
[
!
-f
"./bitsandbytes/libbitsandbytes.so"
]
;
then
# Control will enter here if $DIRECTORY doesn't exist.
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
CUDA_VERSION
=
113 python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
make cleaneggs
export
CUDA_HOME
=
$BASE_PATH
/cuda-11.4
make cuda11x
if
[
!
-f
"./bitsandbytes/libbitsandbytes.so"
]
;
then
# Control will enter here if $DIRECTORY doesn't exist.
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
CUDA_VERSION
=
114 python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
make cleaneggs
export
CUDA_HOME
=
$BASE_PATH
/cuda-11.5
make cuda11x
if
[
!
-f
"./bitsandbytes/libbitsandbytes.so"
]
;
then
# Control will enter here if $DIRECTORY doesn't exist.
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
CUDA_VERSION
=
115 python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
make cleaneggs
export
CUDA_HOME
=
$BASE_PATH
/cuda-11.6
make cuda11x
if
[
!
-f
"./bitsandbytes/libbitsandbytes.so"
]
;
then
# Control will enter here if $DIRECTORY doesn't exist.
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
CUDA_VERSION
=
116 python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
make cleaneggs
export
CUDA_HOME
=
$BASE_PATH
/cuda-11.7
make cuda11x
if
[
!
-f
"./bitsandbytes/libbitsandbytes.so"
]
;
then
# Control will enter here if $DIRECTORY doesn't exist.
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
CUDA_VERSION
=
117 python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
...
...
@@ -128,8 +142,8 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
#
CUDA_VERSION=102-nomatmul python -m build
#
python -m twine upload dist/* --verbose
CUDA_VERSION
=
102-nomatmul python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
...
...
@@ -143,8 +157,8 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
#
CUDA_VERSION=110-nomatmul python -m build
#
python -m twine upload dist/* --verbose
CUDA_VERSION
=
110-nomatmul python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
...
...
@@ -158,8 +172,8 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
#
CUDA_VERSION=111-nomatmul python -m build
#
python -m twine upload dist/* --verbose
CUDA_VERSION
=
111-nomatmul python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
...
...
@@ -172,8 +186,8 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
#
CUDA_VERSION=112-nomatmul python -m build
#
python -m twine upload dist/* --verbose
CUDA_VERSION
=
112-nomatmul python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
...
...
@@ -186,8 +200,8 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
#
CUDA_VERSION=113-nomatmul python -m build
#
python -m twine upload dist/* --verbose
CUDA_VERSION
=
113-nomatmul python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
...
...
@@ -200,8 +214,8 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
#
CUDA_VERSION=114-nomatmul python -m build
##
python -m twine upload dist/* --verbose
CUDA_VERSION
=
114-nomatmul python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
...
...
@@ -214,8 +228,8 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
#
CUDA_VERSION=115-nomatmul python -m build
#
python -m twine upload dist/* --verbose
CUDA_VERSION
=
115-nomatmul python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
...
...
@@ -228,8 +242,8 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
#
CUDA_VERSION=116-nomatmul python -m build
#
python -m twine upload dist/* --verbose
CUDA_VERSION
=
116-nomatmul python
-m
build
python
-m
twine upload dist/
*
--verbose
rm
-rf
dist build
make clean
...
...
@@ -242,5 +256,5 @@ if [ ! -f "./bitsandbytes/libbitsandbytes.so" ]; then
echo
"Compilation unsuccessul!"
1>&2
exit
64
fi
#
CUDA_VERSION=117-nomatmul python -m build
#
python -m twine upload dist/* --verbose
CUDA_VERSION
=
117-nomatmul python
-m
build
python
-m
twine upload dist/
*
--verbose
setup.py
View file @
953b7285
...
...
@@ -14,7 +14,7 @@ version = os.getenv("CUDA_VERSION", "cpu")
setup
(
name
=
"bitsandbytes"
,
version
=
f
"0.
26.0+
{
version
}
"
,
version
=
f
"0.
30.0
"
,
author
=
"Tim Dettmers"
,
author_email
=
"dettmers@cs.washington.edu"
,
description
=
"8-bit optimizers and quantization routines."
,
...
...
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