Skip to content

GitLab

Commit d3e27d8c authored by wooway777's avatar wooway777 Committed by thatPepe
Browse files

demo131 - remove fp32 from paged tests

parent 513a8502
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test/infinicore/ops/paged_attention.py
View file @ d3e27d8c
......@@ -37,7 +37,7 @@ _TOLERANCE_MAP = {
}
# Data types to test
_TENSOR_DTYPES = [infinicore.float16, infinicore.bfloat16, infinicore.float32]
_TENSOR_DTYPES = [infinicore.float16, infinicore.bfloat16]
# ==============================================================================
......
test/infinicore/ops/paged_attention_prefill.py
View file @ d3e27d8c
......@@ -30,7 +30,7 @@ _TOLERANCE_MAP = {
infinicore.bfloat16: {"atol": 2e-2, "rtol": 2e-2},
}
_TENSOR_DTYPES = [infinicore.float16, infinicore.bfloat16, infinicore.float32]
_TENSOR_DTYPES = [infinicore.float16, infinicore.bfloat16]
class SimpleCacheManager:
......
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