Skip to content

Commit

Permalink
Add compile-time architecture support check for bf16 in test_inf
Browse files Browse the repository at this point in the history
…function

Signed-off-by: Leandro Augusto Lacerda Campos <15185896+leandrolcampos@users.noreply.github.com>
  • Loading branch information
leandrolcampos committed Apr 27, 2024
1 parent 33683fd commit d314a48
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions stdlib/test/utils/test_numerics.mojo
Expand Up @@ -53,6 +53,7 @@ fn test_inf() raises:
var msg = "`test_inf` failed for `type == " + str(type) + "`"
assert_true((val > 0.0) & isinf(val), msg=msg)

@parameter
if not has_neon():
# "bf16 is not supported for ARM architectures"
_test_inf[DType.bfloat16]()
Expand Down

0 comments on commit d314a48

Please sign in to comment.