Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add guard for calling gpu implementations of casting storage types #49

Closed
Closed
Changes from 1 commit
Commits
Show all changes
22 commits
Select commit Hold shift + click to select a range
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
replace long with int for python3 (#40)
  • Loading branch information
eric-haibin-lin authored May 19, 2017
commit 2881185b50b2364041c60e4c28d1739600cea1f8
2 changes: 1 addition & 1 deletion python/mxnet/test_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,7 @@ def rand_sparse_ndarray(shape, storage_type, density=None):
# TODO(haibin) support high dim sparse ndarray
assert(len(shape) < 3)
prod = np.prod(shape)
num_cols = long(prod / shape[0])
num_cols = int(prod / shape[0])
# sample index
idx_sample = rnd.rand(shape[0])
indices = np.argwhere(idx_sample < density).flatten()
Expand Down