Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
fix the bug on sparse tensor
Browse files Browse the repository at this point in the history
  • Loading branch information
lanking520 committed Jul 10, 2019
1 parent 6b2b927 commit 63a3cd6
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 9 deletions.
16 changes: 8 additions & 8 deletions ci/docker/runtime_functions.sh
Original file line number Diff line number Diff line change
Expand Up @@ -952,13 +952,6 @@ unittest_ubuntu_python3_quantization_gpu() {
nosetests-3.4 $NOSE_COVERAGE_ARGUMENTS $NOSE_TIMER_ARGUMENTS --with-xunit --xunit-file nosetests_quantization_gpu.xml --verbose tests/python/quantization_gpu
}

unittest_ubuntu_cpu_scala() {
set -ex
scala_prepare
cd scala-package
mvn -B integration-test
}

unittest_centos7_cpu_scala() {
set -ex
cd /work/mxnet
Expand Down Expand Up @@ -1156,12 +1149,19 @@ integrationtest_ubuntu_cpu_dist_kvstore() {
../../tools/launch.py -n 3 --launcher local python test_server_profiling.py
}

integrationtest_ubuntu_cpu_scala() {
set -ex
scala_prepare
cd scala-package
mvn -B verify -DskipTests=false
}

integrationtest_ubuntu_gpu_scala() {
set -ex
scala_prepare
cd scala-package
export SCALA_TEST_ON_GPU=1
mvn -B integration-test -DskipTests=false
mvn -B verify -DskipTests=false
}

integrationtest_ubuntu_gpu_dist_kvstore() {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -956,7 +956,7 @@ class NDArray private[mxnet](private[mxnet] val handle: NDArrayHandle,
DType(mxDtype.value)
}

val sparseFormat: SparseFormat = {
lazy val sparseFormat: SparseFormat = {
val mxSF = new RefInt
checkCall(_LIB.mxNDArrayGetStorageType(handle, mxSF))
SparseFormat(mxSF.value)
Expand Down

0 comments on commit 63a3cd6

Please sign in to comment.