@platforma-open/milaboratories.gpu-test.model 0.1.0

package/eslint.config.mjs

@@ -0,0 +1,4 @@
+import { model } from '@platforma-sdk/eslint-config';
+
+/** @type {import('eslint').Linter.Config[]} */
+export default [...model];

package/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "@platforma-open/milaboratories.gpu-test.model",
+  "version": "0.1.0",
+  "description": "Block model",
+  "type": "module",
+  "main": "dist/index.cjs",
+  "module": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "sources": "./src/index.ts",
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    },
+    "./dist/*": "./dist/*"
+  },
+  "dependencies": {
+    "@platforma-sdk/model": "1.53.15"
+  },
+  "devDependencies": {
+    "@milaboratories/ts-builder": "1.2.9",
+    "@milaboratories/ts-configs": "1.2.1",
+    "@platforma-sdk/block-tools": "2.7.2",
+    "@platforma-sdk/eslint-config": "1.2.0",
+    "eslint": "^9.25.1",
+    "vitest": "^4.0.7",
+    "typescript": "~5.6.3"
+  },
+  "scripts": {
+    "build": "ts-builder build --target block-model && block-tools build-model",
+    "watch": "ts-builder build --target block-model --watch",
+    "type-check": "ts-builder type-check --target block-model",
+    "test": "vitest",
+    "lint": "eslint .",
+    "do-pack": "shx rm -f *.tgz && pnpm pack && shx mv *.tgz package.tgz"
+  }
+}

package/src/index.ts

@@ -0,0 +1,63 @@
+import { BlockModel } from '@platforma-sdk/model';
+
+// eslint-disable-next-line @typescript-eslint/no-empty-object-type -- GPU config will be added in Step 8
+export type BlockArgs = {};
+
+// eslint-disable-next-line @typescript-eslint/no-empty-object-type -- UI state will be added in Step 8
+export type UiState = {};
+
+export const model = BlockModel.create()
+  .withArgs<BlockArgs>({})
+  .withUiState<UiState>({})
+  .argsValid(() => true)
+
+  .output('gpuLog', (ctx) => ctx.outputs?.resolve('gpuLog')?.getLogHandle())
+  .output('gpuInfo', (ctx) => ctx.outputs?.resolve('gpuInfo')?.getDataAsJson<GpuReport>())
+  .output('isRunning', (ctx) => ctx.outputs?.getIsReadyOrError() === false)
+
+  .title(() => 'GPU Test')
+  .sections(() => [{ type: 'link' as const, href: '/' as const, label: 'GPU Info' }])
+  .done(1);
+
+export interface GpuReport {
+  gpu_available: boolean;
+  torch_cuda: {
+    available: boolean;
+    device_count: number;
+    cuda_version: string | null;
+    cudnn_version: string | null;
+    devices: Array<{
+      index: number;
+      name: string;
+      total_memory_mb: number;
+      free_memory_mb?: number;
+      major: number;
+      minor: number;
+      multi_processor_count: number;
+    }>;
+    error?: string;
+  };
+  nvidia_smi: {
+    available: boolean;
+    driver_version: string | null;
+    devices: Array<{
+      index: number;
+      name: string;
+      memory_total_mb: number;
+      memory_free_mb: number;
+      memory_used_mb: number;
+      temperature_c: number | null;
+    }>;
+    error?: string;
+  };
+  environment: Record<string, string>;
+  benchmark: {
+    ran: boolean;
+    matrix_size?: number;
+    cpu_time_ms?: number;
+    gpu_time_ms?: number;
+    speedup?: number;
+    skipped?: string;
+    error?: string;
+  };
+}

package/tsconfig.json

@@ -0,0 +1,9 @@
+{
+  "extends": "@milaboratories/ts-configs/block/model",
+  "compilerOptions": {
+    "outDir": "./dist",
+    "rootDir": "./src"
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist"]
+}