Skip to content

Commit b2ed65d

Browse files
author
Diptorup Deb
committed
Temp commit to add a driver.py to run refactored code base.
1 parent 3060250 commit b2ed65d

File tree

2 files changed

+38
-0
lines changed

2 files changed

+38
-0
lines changed

driver.py

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
#! /usr/bin/env python
2+
3+
# SPDX-FileCopyrightText: 2020 - 2022 Intel Corporation
4+
#
5+
# SPDX-License-Identifier: Apache-2.0
6+
import dpctl.tensor as dpt
7+
8+
import numba_dpex as dpex
9+
from numba_dpex.core.kernel_interface.dispatcher import Dispatcher
10+
11+
12+
# @dpex.kernel
13+
def data_parallel_sum(a, b, c):
14+
"""
15+
Vector addition using the ``kernel`` decorator.
16+
"""
17+
i = dpex.get_global_id(0)
18+
c[i] = a[i] + b[i]
19+
20+
21+
def main():
22+
a = dpt.arange(0, 100, device="level_zero:gpu:0")
23+
b = dpt.arange(0, 100, device="level_zero:gpu:0")
24+
c = dpt.zeros_like(a, device="level_zero:gpu:0")
25+
26+
d = Dispatcher(pyfunc=data_parallel_sum)
27+
d(a, b, c, global_range=[100])
28+
# data_parallel_sum[(10,), (1)](a, b, c)
29+
print(dpt.asnumpy(a))
30+
print(dpt.asnumpy(b))
31+
print(dpt.asnumpy(c))
32+
print("Done...")
33+
34+
35+
if __name__ == "__main__":
36+
main()

numba_dpex/core/kernel_interface/dispatcher.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ def __init__(
4343
self.debug_flags = debug_flags
4444
self.compile_flags = compile_flags
4545
self.kernel_name = pyfunc.__name__
46+
# To be removed
4647
self._global_range = None
4748
self._local_range = None
4849

@@ -355,6 +356,7 @@ def __call__(self, *args, global_range, local_range=None):
355356
local_range (_type_): _description_.
356357
"""
357358
argtypes = [self.typingctx.resolve_argument_type(arg) for arg in args]
359+
breakpoint()
358360

359361
# FIXME: Remove along with __getitem__
360362
global_range, local_range = self._get_ranges(global_range, local_range)

0 commit comments

Comments
 (0)