Skip to content

Commit 8fa1ee1

Browse files
committed
fix: pre-commit
1 parent b9816d1 commit 8fa1ee1

File tree

5 files changed

+9
-1
lines changed

5 files changed

+9
-1
lines changed

numba_cuda/numba/cuda/bf16.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -193,10 +193,12 @@ def exp_ol(a):
193193

194194

195195
if sys.version_info >= (3, 11):
196+
196197
@overload(math.exp2, target="cuda")
197198
def exp2_ol(a):
198199
return _make_unary(a, hexp2)
199200

201+
200202
## Public aliases using Numba/Numpy-style type names
201203
# Floating-point
202204
float32_to_bfloat16 = float2bfloat16

numba_cuda/numba/cuda/cudamath.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ class Math_unary_with_fp16(ConcreteTemplate):
5858
signature(types.float16, types.float16),
5959
]
6060

61+
6162
if sys.version_info >= (3, 11):
6263
Math_unary_with_fp16 = infer_global(math.exp2)(Math_unary_with_fp16)
6364

numba_cuda/numba/cuda/fp16.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
# SPDX-License-Identifier: BSD-2-Clause
3+
34
import sys
45
import numba.core.types as types
56
from numba.cuda._internal.cuda_fp16 import (
@@ -191,6 +192,7 @@ def exp_ol(a):
191192

192193

193194
if sys.version_info >= (3, 11):
195+
194196
@overload(math.exp2, target="cuda")
195197
def exp2_ol(a):
196198
return _make_unary(a, hexp2)

numba_cuda/numba/cuda/tests/cudapy/test_math.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -502,6 +502,7 @@ def test_math_exp(self):
502502
self.unary_template_float64(math_exp, np.exp)
503503
self.unary_template_int64(math_exp, np.exp)
504504
self.unary_template_uint64(math_exp, np.exp)
505+
505506
# ---------------------------------------------------------------------------
506507
# test_math_exp2
507508

@@ -510,7 +511,7 @@ def test_math_exp2(self):
510511
self.unary_template_float32(math_exp2, np.exp2)
511512
self.unary_template_float64(math_exp2, np.exp2)
512513
self.unary_template_int64(math_exp2, np.exp2)
513-
self.unary_template_uint64(math_exp2, np.exp2)
514+
self.unary_template_uint64(math_exp2, np.exp2)
514515

515516
# ---------------------------------------------------------------------------
516517
# test_math_expm1

numba_cuda/numba/cuda/typing/mathdecl.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,11 @@ class Math_unary(ConcreteTemplate):
4444
signature(types.float64, types.float64),
4545
]
4646

47+
4748
if sys.version_info >= (3, 11):
4849
Math_unary = infer_global(math.exp2)(Math_unary)
4950

51+
5052
@infer_global(math.atan2)
5153
class Math_atan2(ConcreteTemplate):
5254
cases = [

0 commit comments

Comments
 (0)