Skip to content

GitLab

Commit 314d4a65 authored by zhuyue's avatar zhuyue
Browse files

Issue/658 - Comment out unsupported float16/bfloat16 tests in isin, mse_loss, and pad.

parent 33d0f769
Hide whitespace changes
Inline Side-by-side
Showing with 21 additions and 8 deletions
test/infinicore/ops/isin.py
View file @ 314d4a65
...@@ -22,12 +22,16 @@ _TEST_CASES_DATA = [ ...@@ -22,12 +22,16 @@ _TEST_CASES_DATA = [
] ]
_TOLERANCE_MAP = { _TOLERANCE_MAP = {
infinicore.float16: {"atol": 0, "rtol": 1e-2}, # infinicore.float16: {"atol": 0, "rtol": 1e-2},
infinicore.float32: {"atol": 0, "rtol": 1e-3}, infinicore.float32: {"atol": 0, "rtol": 1e-3},
infinicore.int32: {"atol": 0, "rtol": 0}, 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(): def parse_test_cases():
......
test/infinicore/ops/mse_loss.py
View file @ 314d4a65
...@@ -21,12 +21,17 @@ _TEST_CASES_DATA = [ ...@@ -21,12 +21,17 @@ _TEST_CASES_DATA = [
] ]
_TOLERANCE_MAP = { _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.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(): def parse_test_cases():
......
test/infinicore/ops/pad.py
View file @ 314d4a65
...@@ -22,12 +22,16 @@ _TEST_CASES_DATA = [ ...@@ -22,12 +22,16 @@ _TEST_CASES_DATA = [
] ]
_TOLERANCE_MAP = { _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.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(): def parse_test_cases():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment