Skip to content

Commit

Permalink
Fixes after initial release
Browse files Browse the repository at this point in the history
- Rename test qmpi -> cgx
- Uncomment dequantization
- Increment package release version
  • Loading branch information
ilmarkov committed Nov 22, 2022
1 parent 5ea5d96 commit fb094d1
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,15 +66,15 @@

setup(name='pytorch_cgx',
packages=['cgx_utils'],
version='0.1.0',
version='0.1.1',
description='pytorch extension adding a backend '
'supporting allreduce of quantized buffers.',
long_description=long_description,
long_description_content_type='text/markdown',
author='Ilia Markov',
author_email='[email protected]',
url='https://github.com/IST-DASLab/torch_cgx/',
download_url="https://github.com/IST-DASLab/torch_cgx/archive/refs/tags/v0.1.0.tar.gz",
download_url="https://github.com/IST-DASLab/torch_cgx/archive/refs/tags/v0.1.1.tar.gz",
ext_modules=[cpp_extension.CUDAExtension('torch_cgx', sources=src,
include_dirs=include_dirs,
extra_compile_args={'cxx': cxx_compile_args, 'nvcc': nvcc_compile_args},
Expand Down
4 changes: 2 additions & 2 deletions src/common/compression/cuda_compression_operations.cu
Original file line number Diff line number Diff line change
Expand Up @@ -794,8 +794,8 @@ void CUDA_dequantize_maxmin(const unsigned char *input_data,
int num_threads = THREADS_PER_BLOCK_DECOMPRESS;
int num_blocks =
BLOCKS_PER_GRID((num_elems + PACK_SIZE - 1) / PACK_SIZE, num_threads);
// DEQUANTIZE1<T, ADD>(input, meta_info, output, num_elems, bucket_size, bits,
// stream, num_blocks, num_threads);
DEQUANTIZE1<T, ADD>(input, meta_info, output, num_elems, bucket_size, bits,
stream, num_blocks, num_threads);
}

/* Functions declarations */
Expand Down
File renamed without changes.

0 comments on commit fb094d1

Please sign in to comment.