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
......@@ -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():
......
......@@ -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():
......
......@@ -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():
......
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