Skip to content

Commit d889afd

Browse files
committed
Transition default GPU data format from dpctl to dpnp
1 parent b8e821c commit d889afd

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

configs/common/sklearn.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
"estimator_params": { "n_jobs": "[REMOVE]" }
2020
},
2121
"data": {
22-
"format": "dpctl",
22+
"format": "dpnp",
2323
"order": "C",
2424
"distributed_split": "rank_based"
2525
},

sklbench/benchmarks/sklearn_estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -363,7 +363,7 @@ def dataframe_function(x):
363363
for i in range(n_batches):
364364
method_instance(x.iloc[i * batch_size : (i + 1) * batch_size])
365365

366-
if "ndarray" in str(type(data_args[0])):
366+
if "array" in str(type(data_args[0])):
367367
return ndarray_function
368368
elif "DataFrame" in str(type(data_args[0])):
369369
return dataframe_function

0 commit comments

Comments
 (0)