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

Commit b49cfce

Browse files
committed
Fix cmake
1 parent ec47b3b commit b49cfce

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -758,7 +758,7 @@ if(USE_TVM_OP)
758758

759759
set(TVM_OP_COMPILE_OPTIONS "-o${CMAKE_CURRENT_BINARY_DIR}/libtvmop.so")
760760
if(CUDA_ARCH_BIN)
761-
set(TVM_OP_COMPILE_OPTIONS "${TVM_OP_COMPILE_OPTIONS} --cuda-arch ${CUDA_ARCH_BIN}")
761+
set(TVM_OP_COMPILE_OPTIONS "${TVM_OP_COMPILE_OPTIONS}" "--cuda-arch" "${CUDA_ARCH_BIN}")
762762
endif()
763763
add_custom_command(TARGET mxnet POST_BUILD
764764
COMMAND ${CMAKE_COMMAND} -E env

tests/python/unittest/test_numpy_op.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,6 @@
2929
from common import assertRaises, with_seed
3030
import random
3131
import scipy.stats as ss
32-
from mxnet.test_utils import verify_generator, gen_buckets_probs_with_ppf, collapse_sum_like
33-
from mxnet.runtime import Features
3432
from mxnet.numpy_op_signature import _get_builtin_op
3533
from mxnet.test_utils import verify_generator, gen_buckets_probs_with_ppf, has_tvm_ops
3634
import platform

0 commit comments

Comments
 (0)