diff options
| author | ivarflakstad <69173633+ivarflakstad@users.noreply.github.com> | 2024-01-10 18:50:30 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-01-10 18:50:30 +0100 |
| commit | d3bdd788cfdcf49b6ea539b77647b82a0b979db0 (patch) | |
| tree | ec3c3eb34d9aef650eafc7699b3401c19fdc92d4 /candle-metal-kernels/src/unary.metal | |
| parent | ae06cb74bb132913b4777cd119b915e665c013bb (diff) | |
| download | candle-d3bdd788cfdcf49b6ea539b77647b82a0b979db0.tar.gz candle-d3bdd788cfdcf49b6ea539b77647b82a0b979db0.tar.bz2 candle-d3bdd788cfdcf49b6ea539b77647b82a0b979db0.zip | |
Use __HAVE_BFLOAT__ to check for bfloat support instead of metal version check (#1540)
Diffstat (limited to 'candle-metal-kernels/src/unary.metal')
| -rw-r--r-- | candle-metal-kernels/src/unary.metal | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-metal-kernels/src/unary.metal b/candle-metal-kernels/src/unary.metal index f95f6ba9..dcf803d8 100644 --- a/candle-metal-kernels/src/unary.metal +++ b/candle-metal-kernels/src/unary.metal @@ -127,7 +127,7 @@ UNARY(id, uint32_t, copy_u32, copy_u32_strided) UNARY(id, int64_t, copy_i64, copy_i64_strided) #endif -#if __METAL_VERSION__ >= 310 +#if defined(__HAVE_BFLOAT__) BFLOAT_UNARY_OP(cos) BFLOAT_UNARY_OP(sin) BFLOAT_UNARY_OP(sqr) |