Skip to content

Commit 3a52974

Browse files
committed
[CUDA] Remove htanh from unsupported math ops for CUDA 12.8
This PR removes htanh from the list of unsupported CUDA half operators, as it is started to be supported since CUDA 12.8. A better option is to check the CUDA version at compile time, while due to the rare use of htanh, the removal seems a simpler way to go with.
1 parent f3bd8e1 commit 3a52974

File tree

2 files changed

+0
-3
lines changed

2 files changed

+0
-3
lines changed

src/target/source/literal/cuda_half_t.h

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,6 @@ static inline __device__ __host__ half HALF_MATH_NAME(half x) { \
317317
#if defined(__CUDA_ARCH__)
318318
#if (__CUDA_ARCH__ >= 530)
319319
CUDA_UNSUPPORTED_HALF_MATH_BINARY(hpow, powf)
320-
CUDA_UNSUPPORTED_HALF_MATH_UNARY(htanh, tanhf)
321320
CUDA_UNSUPPORTED_HALF_MATH_UNARY(htan, tanf)
322321
CUDA_UNSUPPORTED_HALF_MATH_UNARY(hatan, atanf)
323322
CUDA_UNSUPPORTED_HALF_MATH_UNARY(herf, erf)
@@ -358,7 +357,6 @@ static inline __device__ __host__ nv_bfloat16 HALF_MATH_NAME(nv_bfloat16 x) {
358357
}
359358
360359
CUDA_UNSUPPORTED_HALF_MATH_BINARY(hpow, powf)
361-
CUDA_UNSUPPORTED_HALF_MATH_UNARY(htanh, tanhf)
362360
CUDA_UNSUPPORTED_HALF_MATH_UNARY(htan, tanf)
363361
CUDA_UNSUPPORTED_HALF_MATH_UNARY(hatan, atanf)
364362
CUDA_UNSUPPORTED_HALF_MATH_UNARY(herf, erf)

tests/python/meta_schedule/test_meta_schedule_mma_m16n8k8_auto_tensorization.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -717,7 +717,6 @@ class TVM_ALIGNED(2) half {
717717
#if defined(__CUDA_ARCH__)
718718
#if (__CUDA_ARCH__ >= 530)
719719
CUDA_UNSUPPORTED_HALF_MATH_BINARY(hpow, powf)
720-
CUDA_UNSUPPORTED_HALF_MATH_UNARY(htanh, tanhf)
721720
CUDA_UNSUPPORTED_HALF_MATH_UNARY(htan, tanf)
722721
CUDA_UNSUPPORTED_HALF_MATH_UNARY(hatan, atanf)
723722
CUDA_UNSUPPORTED_HALF_MATH_UNARY(herf, erf)

0 commit comments

Comments
 (0)