Merge pull request #163 from sk-ai-net/feature/2-mnist-exec

Add initial version of CPU compute backend

Author: michalharakal

settings.gradle.kts

@@ -19,4 +19,7 @@ include("skainet-lang:skainet-lang-core")
 include("skainet-lang:skainet-lang-models")
 include("skainet-lang:skainet-lang-ksp-annotations")
 include("skainet-lang:skainet-lang-ksp-processor")
-include("skainet-lang:skainet-lang-export-ops")
+include("skainet-compile:skainet-compile-core")
+//include("skainet-compile:skainet-compile-dag")
+//include("skainet-backends:skainet-backends-core")
+include("skainet-backends:skainet-backend-cpu")

skainet-backends/skainet-backend-cpu/build.gradle.kts

@@ -0,0 +1,60 @@
+import org.jetbrains.kotlin.gradle.ExperimentalKotlinGradlePluginApi
+import org.jetbrains.kotlin.gradle.ExperimentalWasmDsl
+import org.jetbrains.kotlin.gradle.dsl.JvmTarget
+
+plugins {
+    alias(libs.plugins.kotlinMultiplatform)
+    alias(libs.plugins.androidLibrary)
+    alias(libs.plugins.vanniktech.mavenPublish)
+    alias(libs.plugins.kover)
+    alias(libs.plugins.binary.compatibility.validator)
+}
+
+kotlin {
+    explicitApi()
+
+    androidTarget {
+        @OptIn(ExperimentalKotlinGradlePluginApi::class)
+        compilerOptions {
+            jvmTarget.set(JvmTarget.JVM_11)
+        }
+    }
+
+    iosArm64()
+    iosSimulatorArm64()
+    macosArm64 ()
+    linuxX64 ()
+    linuxArm64 ()
+
+    jvm()
+
+    @OptIn(ExperimentalWasmDsl::class)
+    wasmJs {
+        browser()
+        binaries.executable()
+    }
+
+    sourceSets {
+        commonMain.dependencies {
+            implementation(project(":skainet-lang:skainet-lang-core"))
+            implementation(project(":skainet-compile:skainet-compile-core"))
+        }
+
+        commonTest.dependencies {
+            implementation(libs.kotlin.test)
+        }
+    }
+}
+
+android {
+    namespace = "sk.ai.net.lang.api"
+    compileSdk = libs.versions.android.compileSdk.get().toInt()
+
+    defaultConfig {
+        minSdk = libs.versions.android.minSdk.get().toInt()
+    }
+    compileOptions {
+        sourceCompatibility = JavaVersion.VERSION_11
+        targetCompatibility = JavaVersion.VERSION_11
+    }
+}

skainet-backends/skainet-backend-cpu/gradle.properties

@@ -0,0 +1,2 @@
+POM_ARTIFACT_ID=skainet-backend-cpu
+POM_NAME=skainet neural network scripting API

skainet-backends/skainet-backend-cpu/src/commonMain/kotlin/sk/ainet/context/DirectCpuExecutionContext.kt

@@ -0,0 +1,24 @@
+package sk.ainet.context
+
+import sk.ainet.lang.tensor.data.TensorDataFactory
+import sk.ainet.lang.tensor.data.DenseTensorDataFactory
+import sk.ainet.lang.tensor.ops.TensorOps
+import sk.ainet.exec.tensor.ops.DefaultCpuOps
+
+public class DirectCpuExecutionContext(
+    override val executionStats: ExecutionStats = ExecutionStats(),
+) : ExecutionContext {
+    private val _memoryInfo = MemoryInfo(
+        totalMemory = 0,
+        usedMemory = 0,
+        freeMemory = 0,
+        usagePercentage = 0.0
+    )
+
+    override val tensorDataFactory: TensorDataFactory = DenseTensorDataFactory()
+    override val memoryInfo: MemoryInfo
+        get() = _memoryInfo
+
+    override val ops: TensorOps
+        get() = DefaultCpuOps(tensorDataFactory)
+}