attempt to compile unary functions for CPU only

src/CMakeLists.txt

@@ -27,6 +27,7 @@ set(legateboost_srcs
   models/tree/build_tree.cc
   models/tree/update_tree.cc
   models/tree/predict.cc
+  models/krr/rbf.cc
   models/nn/build_nn.cc
   cpp_utils/cpp_utils.h
   cpp_utils/cpp_utils.cc

src/cpp_utils/cpp_utils.h

@@ -263,7 +263,9 @@ class UnaryOpTask : public Task<UnaryOpTask<F, OpCode>, OpCode> {
     auto const& in = context.input(0);
     legate::dim_dispatch(in.dim(), DispatchDimOp{}, context, in, thrust::host);
   }
+#ifdef LEGATEBOOST_USE_CUDA
   static void gpu_variant(legate::TaskContext context);
+#endif
 };
 
 }  // namespace legateboost

src/models/krr/rbf.cc

@@ -0,0 +1,26 @@
+
+/* Copyright 2024 NVIDIA Corporation
+ *
+ * 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 "rbf.h"
+namespace  // unnamed
+{
+#ifndef LEGATEBOOST_USE_CUDA
+static void __attribute__((constructor)) register_tasks(void)
+{
+  legateboost::RbfTask::register_variants();
+}
+#endif
+}  // namespace

src/special/special.cc

@@ -0,0 +1,28 @@
+/* Copyright 2023-2024, NVIDIA Corporation
+ *
+ * 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 "special.h"
+namespace  // unnamed
+{
+#ifndef LEGATEBOOST_USE_CUDA
+static void __attribute__((constructor)) register_tasks(void)
+{
+  legateboost::ErfTask::register_variants();
+  legateboost::LgammaTask::register_variants();
+  legateboost::TgammaTask::register_variants();
+  legateboost::DigammaTask::register_variants();
+  legateboost::ZetaTask::register_variants();
+}
+#endif
+}  // namespace