bump to cuTENSOR v1.6.2
@@ -108,11 +108,12 @@ def __make_cutensor_record(
def _make_cutensor_record(cuda_version):
return __make_cutensor_record(
- cuda_version, '1.6.1',
- 'libcutensor-linux-x86_64-1.6.1.5-archive.tar.xz',
- 'libcutensor-windows-x86_64-1.6.1.5-archive.zip')
+ cuda_version, '1.6.2',
+ 'libcutensor-linux-x86_64-1.6.2.3-archive.tar.xz',
+ 'libcutensor-windows-x86_64-1.6.2.3-archive.zip')
+_cutensor_records.append(_make_cutensor_record('12.x'))
_cutensor_records.append(_make_cutensor_record('11.x')) # CUDA 11.2+
_cutensor_records.append(_make_cutensor_record('11.1'))
_cutensor_records.append(_make_cutensor_record('11.0'))
