Skip to content

Commit 4805f9c

Browse files
author
Diptorup Deb
committed
Update type annotation to work on Py 3.9
1 parent b653dfa commit 4805f9c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

numba_dpex/experimental/launcher.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22
#
33
# SPDX-License-Identifier: Apache-2.0
44

5+
from typing import Union
6+
57
from llvmlite import ir as llvmir
68
from numba.core import cgutils, cpu, types
79
from numba.extending import intrinsic, overload
@@ -86,7 +88,7 @@ def _create_kernel_launcher_body(
8688
codegen_targetctx: cpu.CPUContext,
8789
kernel_targetctx: DpexKernelTargetContext,
8890
builder: llvmir.IRBuilder,
89-
indexer_argty: RangeType | NdRangeType,
91+
indexer_argty: Union[RangeType, NdRangeType],
9092
kernel_argtys: tuple[types.Type, ...],
9193
kernel_module: _KernelModule,
9294
index_space_arg: llvmir.BaseStructType,

0 commit comments

Comments
 (0)