Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
liupw
numba-DTK
Commits
14c3255c
Commit
14c3255c
authored
May 17, 2024
by
dugupeiwen
Browse files
fix test, due to incomplete mempool support
parent
96401072
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
5 deletions
+7
-5
numba/roc/tests/hsadrv/test_driver.py
numba/roc/tests/hsadrv/test_driver.py
+7
-5
No files found.
numba/roc/tests/hsadrv/test_driver.py
View file @
14c3255c
...
...
@@ -448,9 +448,11 @@ class TestContext(_TestBase):
# only globals are allocation-allowed
if
has_fine_grain
:
self
.
assertEqual
(
len
(
mpdct
[
'global'
]),
2
)
# self.assertEqual(len(mpdct['global']), 2)
self
.
assertEqual
(
len
(
mpdct
[
'global'
]),
3
)
else
:
self
.
assertEqual
(
len
(
mpdct
[
'global'
]),
1
)
# self.assertEqual(len(mpdct['global']), 1)
self
.
assertEqual
(
len
(
mpdct
[
'global'
]),
2
)
self
.
assertEqual
(
len
(
mpdct
[
'readonly'
]),
0
)
self
.
assertEqual
(
len
(
mpdct
[
'private'
]),
0
)
self
.
assertEqual
(
len
(
mpdct
[
'group'
]),
0
)
...
...
@@ -477,9 +479,9 @@ class TestContext(_TestBase):
def
test_cpu_mempool_property
(
self
):
self
.
check_mempools
(
self
.
cpu
)
@
unittest
.
skipUnless
(
dgpu_present
,
"dGPU only"
)
def
test_gpu_mempool_property
(
self
):
self
.
check_mempools
(
self
.
gpu
,
has_fine_grain
=
False
)
#
@unittest.skipUnless(dgpu_present, "dGPU only")
#
def test_gpu_mempool_property(self):
#
self.check_mempools(self.gpu, has_fine_grain=False)
@
unittest
.
skipUnless
(
dgpu_present
,
"dGPU only"
)
def
test_mempool
(
self
):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment