adding a method to add your own cpp functions to be executed over pytaskr

Author: noabauma

examples/local/meson.build

@@ -10,4 +10,5 @@ subdir('pendingOperation')
 subdir('workerSpecific')
 subdir('manyParallel')
 subdir('suspend')
-subdir('simple')
+subdir('simple')
+subdir('mmm')

examples/local/mmm/README.rst

@@ -0,0 +1,7 @@
+ABC Tasks
+============
+
+A simple example for creating a task dependency-based runtime system using HiCR
+
+It creates and adds tasks in an arbitrary order. However, given the dependencies should force the output to always be an ordered sequence of A->B->C. This example makes sure TaskR is handling these dependencies correctly.
+

examples/local/mmm/meson.build

@@ -0,0 +1,11 @@
+testSuite = [ 'examples', 'local', 'mmmm' ]
+
+if get_option('buildPyTaskR') and get_option('buildTests')
+	test('pyTaskR', 
+		py,
+		args : [ 'py_source/main.py' ],
+		is_parallel : false, 
+		env: ['PYTHONPATH=' + meson.project_build_root() + '/include/pytaskr/'],
+		suite: testSuite, 
+		workdir: meson.current_source_dir())
+endif

examples/local/mmm/py_source/main.py

@@ -0,0 +1,33 @@
+"""
+    Copyright 2025 Huawei Technologies Co., Ltd.
+ 
+  Licensed under the Apache License, Version 2.0 (the "License");
+  you may not use this file except in compliance with the License.
+  You may obtain a copy of the License at
+ 
+      http://www.apache.org/licenses/LICENSE-2.0
+ 
+  Unless required by applicable law or agreed to in writing, software
+  distributed under the License is distributed on an "AS IS" BASIS,
+  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+  See the License for the specific language governing permissions and
+  limitations under the License.
+"""
+
+import taskr
+# from mmm import mmmDriver
+from mmm_cpp import mmm_cpp_Driver
+
+def main():
+    # Initialize taskr with the wanted compute manager backend and number of PUs
+    t = taskr.taskr(taskr.HiCRBackend.nosv, 2)
+
+    # Get the runtime
+    runtime = t.get_runtime()
+
+    # Running mmm example
+    # mmmDriver(runtime)
+    mmm_cpp_Driver(runtime)
+
+if __name__ == "__main__":
+    main()

examples/local/mmm/py_source/mmm.py

@@ -0,0 +1,63 @@
+"""
+    Copyright 2025 Huawei Technologies Co., Ltd.
+ 
+  Licensed under the Apache License, Version 2.0 (the "License");
+  you may not use this file except in compliance with the License.
+  You may obtain a copy of the License at
+ 
+      http://www.apache.org/licenses/LICENSE-2.0
+ 
+  Unless required by applicable law or agreed to in writing, software
+  distributed under the License is distributed on an "AS IS" BASIS,
+  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+  See the License for the specific language governing permissions and
+  limitations under the License.
+"""
+
+import time
+import numpy as np
+import taskr
+
+NTASKS = 2
+
+N = 1000
+
+def mmm(task):
+  A = np.zeros((N,N))
+  B = np.empty((N,N))
+  C = np.empty((N,N))
+
+  for i in range(N):
+    for j in range(N):
+      B[i, j] = i
+      C[i, j] = j
+
+  B += task.getLabel()+1
+  C += task.getLabel()+1
+  
+  A = B @ C
+  print(f"Hello, I am task {task.getLabel()}")
+
+
+def mmmDriver(runtime):
+  # Initializing taskr
+  runtime.initialize()
+
+  taskfc = taskr.Function(mmm)
+
+  # Adding to tasks to taskr
+  for i in range(NTASKS):
+    runtime.addTask(taskr.Task(i, taskfc, i))
+
+  # Running taskr for the current repetition
+  t_start = time.time()
+  runtime.run()
+
+  # Waiting current repetition to end
+  runtime.await_()
+  t_duration = time.time() - t_start
+
+  print(f"total time: {t_duration}s")
+
+  # Finalizing taskr
+  runtime.finalize()

examples/local/mmm/py_source/mmm_cpp.py

@@ -0,0 +1,41 @@
+"""
+    Copyright 2025 Huawei Technologies Co., Ltd.
+ 
+  Licensed under the Apache License, Version 2.0 (the "License");
+  you may not use this file except in compliance with the License.
+  You may obtain a copy of the License at
+ 
+      http://www.apache.org/licenses/LICENSE-2.0
+ 
+  Unless required by applicable law or agreed to in writing, software
+  distributed under the License is distributed on an "AS IS" BASIS,
+  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+  See the License for the specific language governing permissions and
+  limitations under the License.
+"""
+
+import time
+import taskr
+
+NTASKS = 2
+
+def mmm_cpp_Driver(runtime):
+  # Initializing taskr
+  runtime.initialize()
+
+  taskfc = taskr.get_cpp_function("cpp_mmm")
+
+  # Adding to tasks to taskr
+  for i in range(NTASKS):
+    runtime.addTask(taskr.Task(i, taskfc))
+
+  # Running taskr for the current repetition
+  t_start = time.time()
+  runtime.run()
+
+  # Waiting current repetition to end
+  runtime.await_()
+  print(f"total time: {time.time() - t_start}")
+
+  # Finalizing taskr
+  runtime.finalize()

examples/local/simple/py_source/simple.py

@@ -13,10 +13,6 @@
   See the License for the specific language governing permissions and
   limitations under the License.
 """
-import ctypes
-
-libc = ctypes.CDLL("libc.so.6")
-libc.sched_getcpu.restype = ctypes.c_int
 
 import taskr
 

extern/hicr

@@ -6,7 +6,7 @@ py = import('python').find_installation(pure: false)
 pybind11_dep = dependency('pybind11', required: true)
 
 py.extension_module('taskr',
-    'pytaskr.cpp',
+    ['pytaskr.cpp', 'mmm.cpp'],
     install: true,
     dependencies : [TaskRBuildDep, pybind11_dep],
 )

include/pytaskr/meson.build

@@ -0,0 +1,67 @@
+/*
+ *   Copyright 2025 Huawei Technologies Co., Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <chrono>
+
+#include <taskr/taskr.hpp>
+#include <pytaskr/pytaskr.hpp>
+
+#define mytype float
+
+#define NTASKS 2
+
+/**
+ * Compute mmm locally
+ */
+void mmm(taskr::Task *) {
+  const size_t N = 1000;
+
+  // Allocate memory
+  volatile mytype *A = (mytype *)calloc(1, N * N * sizeof(mytype));
+  volatile mytype *B = (mytype *)malloc(N * N * sizeof(mytype));
+  volatile mytype *C = (mytype *)malloc(N * N * sizeof(mytype));
+
+  // Filling matrices B and C
+  for (size_t i = 0; i < N; ++i) {
+    for (size_t j = 0; j < N; ++j) {
+      B[i * N + j] = 1.0/(mytype(i + 1));
+      C[i * N + j] = 1.0/(mytype(j + 1));
+    }
+  }
+
+  // mmm
+  for (size_t i = 0; i < N; ++i) {
+    for (size_t j = 0; j < N; ++j) {
+      for (size_t k = 0; k < N; ++k) {
+        A[i * N + j] += B[i * N + k] * C[k * N + j];
+      }
+    }
+  }
+
+  // free memory
+  free((mytype*)A);
+  free((mytype*)B);
+  free((mytype*)C);
+}
+
+namespace taskr {
+
+__attribute__((constructor))  // GCC/Clang: Run before main/init
+void register_my_func() {
+    register_function("cpp_mmm", mmm);
+}
+
+}