Commit 027b9655 authored by yan.yan's avatar yan.yan
Browse files

v2.1.15: fix missing Hash in setup.py in 2.1.14

parent 076bdb05
# Changelog # Changelog
## [2.1.15] - 2021-11-28
### Fixed
- Fix missing pccm.Class in setup.py
## [2.1.14] - 2021-11-28 ## [2.1.14] - 2021-11-28
### Added ### Added
- Add hash table - Add hash table
......
...@@ -157,6 +157,7 @@ if disable_jit is not None and disable_jit == "1": ...@@ -157,6 +157,7 @@ if disable_jit is not None and disable_jit == "1":
from cumm.constants import CUMM_CPU_ONLY_BUILD from cumm.constants import CUMM_CPU_ONLY_BUILD
from spconv.csrc.sparse.all import SpconvOps from spconv.csrc.sparse.all import SpconvOps
from spconv.csrc.utils import BoxOps from spconv.csrc.utils import BoxOps
from spconv.csrc.hash.core import HashTable
cu = GemmMainUnitTest(SHUFFLE_SIMT_PARAMS + SHUFFLE_VOLTA_PARAMS + SHUFFLE_TURING_PARAMS) cu = GemmMainUnitTest(SHUFFLE_SIMT_PARAMS + SHUFFLE_VOLTA_PARAMS + SHUFFLE_TURING_PARAMS)
convcu = ConvMainUnitTest(IMPLGEMM_SIMT_PARAMS + IMPLGEMM_VOLTA_PARAMS + IMPLGEMM_TURING_PARAMS) convcu = ConvMainUnitTest(IMPLGEMM_SIMT_PARAMS + IMPLGEMM_VOLTA_PARAMS + IMPLGEMM_TURING_PARAMS)
...@@ -170,9 +171,9 @@ if disable_jit is not None and disable_jit == "1": ...@@ -170,9 +171,9 @@ if disable_jit is not None and disable_jit == "1":
std = "c++14" std = "c++14"
else: else:
std = "c++17" std = "c++17"
cus = [cu, convcu, SpconvOps(), BoxOps()] cus = [cu, convcu, SpconvOps(), BoxOps(), HashTable()]
if CUMM_CPU_ONLY_BUILD: if CUMM_CPU_ONLY_BUILD:
cus = [SpconvOps(), BoxOps()] cus = [SpconvOps(), BoxOps(), HashTable()]
ext_modules: List[Extension] = [ ext_modules: List[Extension] = [
PCCMExtension(cus, PCCMExtension(cus,
"spconv/core_cc", "spconv/core_cc",
......
2.1.14 2.1.15
\ No newline at end of file \ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment