Skip to content

Commit

Permalink
Update type annotation to work on Py 3.9
Browse files Browse the repository at this point in the history
  • Loading branch information
Diptorup Deb committed Oct 26, 2023
1 parent cdb8da9 commit eacaa3a
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion numba_dpex/experimental/launcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
#
# SPDX-License-Identifier: Apache-2.0

from typing import Union

from llvmlite import ir as llvmir
from numba.core import cgutils, cpu, types
from numba.extending import intrinsic, overload
Expand Down Expand Up @@ -86,7 +88,7 @@ def _create_kernel_launcher_body(
codegen_targetctx: cpu.CPUContext,
kernel_targetctx: DpexKernelTargetContext,
builder: llvmir.IRBuilder,
indexer_argty: RangeType | NdRangeType,
indexer_argty: Union[RangeType, NdRangeType],
kernel_argtys: tuple[types.Type, ...],
kernel_module: _KernelModule,
index_space_arg: llvmir.BaseStructType,
Expand Down

0 comments on commit eacaa3a

Please sign in to comment.