-
Notifications
You must be signed in to change notification settings - Fork 547
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
[REVIEW] FIL benchmark now works again with gpuarray-c input type #2209
Conversation
Please update the changelog in order to start CI tests. View the gpuCI docs here. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
More of a question than a review below...
Also, is there a quick test we can expand to use gpuarray-c so it doesn't break again?
else: | ||
raise TypeError("Received unsupported input type") | ||
# convert the input if necessary | ||
y_pred1 = (y_pred.copy_to_host() if cuda.devicearray.is_cuda_ndarray(y_pred) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why do these need to be host arrays? Seems like it should work with gpu arrays too
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I do agree that the new code is simpler, though
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It doesn't have to be a host array. However, it needs to be an array type that behaves like a numpy array, e.g. supports the >
operator. Also, this part is not performance-critical, so a host array should be fine.
Any specific suggestions for a GPU array and a standard way to convert into it?
Added a test. Could you take another look? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
One minor suggestion, but pre-approving as it all LGTM
gpuarray-c
input type_treelite_fil_accuracy_score