Skip to content
This repository has been archived by the owner on Jul 2, 2021. It is now read-only.

Commit

Permalink
Merge pull request #332 from Hakuyume/support-cupy-v2
Browse files Browse the repository at this point in the history
Support cupy v2
  • Loading branch information
yuyu2172 authored Jul 12, 2017
2 parents 2da817f + 72e4328 commit c2f39d5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion chainercv/utils/bbox/non_maximum_suppression.py
Original file line number Diff line number Diff line change
Expand Up @@ -196,7 +196,7 @@ def _call_nms_kernel(bbox, thresh):
mask_dev = cp.zeros((n_bbox * col_blocks,), dtype=np.uint64)
bbox = cp.ascontiguousarray(bbox, dtype=np.float32)
kern = _load_kernel('nms_kernel', _nms_gpu_code)
kern(blocks, threads, args=(n_bbox, cp.float32(thresh),
kern(blocks, threads, args=(cp.int32(n_bbox), cp.float32(thresh),
bbox, mask_dev))

mask_host = mask_dev.get()
Expand Down

0 comments on commit c2f39d5

Please sign in to comment.