We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f92751e commit b9816d1Copy full SHA for b9816d1
numba_cuda/numba/cuda/fp16.py
@@ -1,6 +1,6 @@
1
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2
# SPDX-License-Identifier: BSD-2-Clause
3
-
+import sys
4
import numba.core.types as types
5
from numba.cuda._internal.cuda_fp16 import (
6
typing_registry,
@@ -190,9 +190,10 @@ def exp_ol(a):
190
return _make_unary(a, hexp)
191
192
193
-@overload(math.exp2, target="cuda")
194
-def exp2_ol(a):
195
- return _make_unary(a, hexp2)
+if sys.version_info >= (3, 11):
+ @overload(math.exp2, target="cuda")
+ def exp2_ol(a):
196
+ return _make_unary(a, hexp2)
197
198
199
@overload(math.tanh, target="cuda")
0 commit comments