|
|
@@ -414,7 +414,7 @@ class InferImpl : public Infer {
|
|
|
for (int ib = 0; ib < num_image; ++ib) {
|
|
|
float *output_array_device = output_array_.gpu() + ib * num_classes_;
|
|
|
float *classes_indices_device = classes_indices_.gpu() + ib;
|
|
|
- classfier_softmax(output_array_device, num_classes_, stream_, classes_indices_device);
|
|
|
+ classfier_softmax(output_array_device, num_classes_, classes_indices_device, stream_);
|
|
|
}
|
|
|
|
|
|
|
