Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
jerrrrry
infinicore
Commits
314d4a65
Commit
314d4a65
authored
Nov 22, 2025
by
zhuyue
Browse files
Issue/658 - Comment out unsupported float16/bfloat16 tests in isin, mse_loss, and pad.
parent
33d0f769
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
21 additions
and
8 deletions
+21
-8
test/infinicore/ops/isin.py
test/infinicore/ops/isin.py
+6
-2
test/infinicore/ops/mse_loss.py
test/infinicore/ops/mse_loss.py
+8
-3
test/infinicore/ops/pad.py
test/infinicore/ops/pad.py
+7
-3
No files found.
test/infinicore/ops/isin.py
View file @
314d4a65
...
...
@@ -22,12 +22,16 @@ _TEST_CASES_DATA = [
]
_TOLERANCE_MAP
=
{
infinicore
.
float16
:
{
"atol"
:
0
,
"rtol"
:
1e-2
},
#
infinicore.float16: {"atol": 0, "rtol": 1e-2},
infinicore
.
float32
:
{
"atol"
:
0
,
"rtol"
:
1e-3
},
infinicore
.
int32
:
{
"atol"
:
0
,
"rtol"
:
0
},
}
_TENSOR_DTYPES
=
[
infinicore
.
int32
,
infinicore
.
float32
,
infinicore
.
float16
]
_TENSOR_DTYPES
=
[
infinicore
.
int32
,
infinicore
.
float32
,
# infinicore.float16,
]
def
parse_test_cases
():
...
...
test/infinicore/ops/mse_loss.py
View file @
314d4a65
...
...
@@ -21,12 +21,17 @@ _TEST_CASES_DATA = [
]
_TOLERANCE_MAP
=
{
infinicore
.
float16
:
{
"atol"
:
1e-3
,
"rtol"
:
1e-2
},
#
infinicore.float16: {"atol": 1e-3, "rtol": 1e-2},
infinicore
.
float32
:
{
"atol"
:
1e-5
,
"rtol"
:
1e-4
},
infinicore
.
bfloat16
:
{
"atol"
:
1e-2
,
"rtol"
:
5e-2
},
#
infinicore.bfloat16: {"atol": 1e-2, "rtol": 5e-2},
}
_TENSOR_DTYPES
=
[
infinicore
.
float16
,
infinicore
.
bfloat16
,
infinicore
.
float32
]
_TENSOR_DTYPES
=
[
# infinicore.float16,
# some pytorch version doesn't support bfloat16
# infinicore.bfloat16,
infinicore
.
float32
,
]
def
parse_test_cases
():
...
...
test/infinicore/ops/pad.py
View file @
314d4a65
...
...
@@ -22,12 +22,16 @@ _TEST_CASES_DATA = [
]
_TOLERANCE_MAP
=
{
infinicore
.
float16
:
{
"atol"
:
1e-2
,
"rtol"
:
1e-2
},
#
infinicore.float16: {"atol": 1e-2, "rtol": 1e-2},
infinicore
.
float32
:
{
"atol"
:
1e-5
,
"rtol"
:
1e-4
},
infinicore
.
bfloat16
:
{
"atol"
:
1e-2
,
"rtol"
:
5e-2
},
#
infinicore.bfloat16: {"atol": 1e-2, "rtol": 5e-2},
}
_TENSOR_DTYPES
=
[
infinicore
.
float16
,
infinicore
.
bfloat16
,
infinicore
.
float32
]
_TENSOR_DTYPES
=
[
# infinicore.float16,
# infinicore.bfloat16,
infinicore
.
float32
,
]
def
parse_test_cases
():
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment