Skip to content

Commit a8f4ca6

Browse files
Sync pykokkos-base with kokkos version 3.7.00 commit d19aab9 (#49)
* Sync pykokkos-base with kokkos version 3.7.00 commit d19aab9 * MAINT: PR 39 revisions * reduce compiled view ranks from 5 to 4 in attempt to alleviate compilation memory issue * the reason for going above 3 is that 4 is the current preferred minimum for the Python array API standard per this discussion: data-apis/array-api#479 * Tests: add add kokkos.finalize() to tearDownClass() * CI: disable multiple cores when building pykokkos-base in python-build action to avoid running out of memory * CI: bind kokkos.is_finalized() and guard finalize() call in tearDownClass * formatting Co-authored-by: Tyler Reddy <[email protected]>
1 parent e94d08e commit a8f4ca6

File tree

4 files changed

+6
-3
lines changed

4 files changed

+6
-3
lines changed

.github/workflows/linux-ci.yml

+2-1
Original file line numberDiff line numberDiff line change
@@ -119,10 +119,11 @@ jobs:
119119
python -m pip install pytest &&
120120
python setup.py install
121121
--enable-layouts
122-
--enable-view-ranks=5
122+
--enable-view-ranks=4
123123
--disable-memory-traits
124124
--enable-werror
125125
-- -DENABLE_EXAMPLES=ON
126+
-- -j1
126127

127128
- name: Import Test
128129
run:

external/kokkos

Submodule kokkos updated 846 files

kokkos/test/views.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,8 @@ def setUpClass(self):
7070

7171
@classmethod
7272
def tearDownClass(self):
73-
pass
73+
if not kokkos.is_finalized():
74+
kokkos.finalize()
7475

7576
def setUp(self):
7677
pass

src/libpykokkos.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,7 @@ PYBIND11_MODULE(libpykokkos, kokkos) {
103103

104104
kokkos.def("is_initialized", &Kokkos::is_initialized,
105105
"Query initialization state");
106+
kokkos.def("is_finalized", &Kokkos::is_finalized, "Query finalization state");
106107
kokkos.def("initialize", _initialize, "Initialize Kokkos");
107108
kokkos.def("finalize", _finalize, "Finalize Kokkos");
108109

0 commit comments

Comments
 (0)