Skip to content

Commit 4f65e1f

Browse files
add bf16 cases
1 parent a4653a1 commit 4f65e1f

File tree

2 files changed

+55
-0
lines changed

2 files changed

+55
-0
lines changed
Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
{
2+
"INCLUDE": ["../../common/sklearn.json"],
3+
"PARAMETERS_SETS": {
4+
"basic stats parameters": {
5+
"algorithm": {
6+
"estimator": "BasicStatistics"
7+
},
8+
"data": {
9+
"dtype": ["float32"]
10+
}
11+
},
12+
"synthetic data": {
13+
"data": [
14+
{ "source": "make_blobs", "generation_kwargs": { "n_samples": 10000000, "n_features": 10, "centers": 1 } }
15+
]
16+
}
17+
},
18+
"TEMPLATES": {
19+
"basic_statistics": {
20+
"SETS": [
21+
"sklearn-ex[gpu] implementations",
22+
"basic stats parameters",
23+
"synthetic data"
24+
]
25+
}
26+
}
27+
}

configs/regular/bf16/covariance.json

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
{
2+
"INCLUDE": ["../../common/sklearn.json"],
3+
"PARAMETERS_SETS": {
4+
"covariance parameters": {
5+
"algorithm": {
6+
"estimator": "EmpiricalCovariance",
7+
"library": "sklearnex.preview.covariance"
8+
},
9+
"data": {
10+
"dtype": ["float32"]
11+
}
12+
},
13+
"synthetic data": {
14+
"data": [
15+
{ "source": "make_blobs", "generation_kwargs": { "n_samples": 10000000, "n_features": 10, "centers": 1 } }
16+
]
17+
}
18+
},
19+
"TEMPLATES": {
20+
"covariance": {
21+
"SETS": [
22+
"sklearn-ex[gpu] implementations",
23+
"covariance parameters",
24+
"synthetic data"
25+
]
26+
}
27+
}
28+
}

0 commit comments

Comments
 (0)