resource deallocation (debugging WIP)

Author: austinvhuang

.github/workflows/build.yml

@@ -0,0 +1,14 @@
+name: build
+
+on:
+  push:
+  pull_request:
+    types: [opened, reopened, labeled, unlabeled, synchronize]
+  workflow_dispatch:
+
+jobs:
+  build:
+    - name: Build
+      run: 
+        echo "Hello, ${{ github.actor }}"
+      

Makefile

@@ -66,3 +66,9 @@ clean-build:
 
 clean:
 	read -r -p "This will delete the contents of build/* and third_party/*. Are you sure? [CTRL-C to abort] " response && rm -rf build/* third_party/fetchcontent/*
+
+all: build
+	cd examples/gpu_puzzles && make
+	cd examples/hello_world && make
+	cd eamples/raymarch && make
+	cd examples/webgpu_intro && make

README.md

@@ -23,7 +23,7 @@ gpu.cpp provides a minimal set of composable functions and types that make
 WebGPU compute simple and concise to work with for GPU compute R&D use cases,
 while keeping abstractions minimal and transparent.
 
-# Hello World: A GELU Kernel
+## Hello World: A GELU Kernel
 
 Here's an GELU kernel implemented (based on the CUDA implementation of
 [llm.c](https://github.com/karpathy/llm.c)) as an on-device WGSL shader and
@@ -80,7 +80,7 @@ For those curious about what happens under the hood with the raw WebGPU API,
 the equivalent functionality is implemented using the WebGPU C API in
 `examples/webgpu_intro/run.cpp`.
 
-## Quick Start
+## Quick Start: Building and Running
 
 The only dependency of this library is a WebGPU implementation. Currently we
 recommend using the Dawn backend until further testing, but we plan to support
@@ -115,18 +115,8 @@ Welcome!
 
 This program is a brief intro to the gpu.cpp library.
 
-You can use the library by simply including the gpu.h header, starting with a
-build template (see examples/hello_gpu/ for a template project that builds the
-library).
-
-  #include "gpu.h"
-
-See `examples/hello_world/` for an examle of build scripts to run a standalone
-program that uses this library.
+...
 
-┌──────────────────────────────┐
-│ Press Enter to Continue...   │
-└──────────────────────────────┘
 ```
 
 The first time you build and run the project, it will download the WebGPU

examples/README.md

@@ -1,14 +1,14 @@
 # Examples (Work-in-Progress)
 
-NOTE: These examples are a work-in-progress - do not use them yet.
-
 Here are some standalone projects exemplifying how to use this library as well
-as what's under the hood.
+as what's under the hood. NOTE: These examples are a work-in-progress.
+
 
 In order of expertise (beginner to advanced):
 
 - hello_world - Minimal template project to get started with the gpu.cpp library.
-- raymarch (TODO) - Using the webgpu computation to implement a toy raymarcher
+- gpu_puzzles - (WIP) Implementation of Sasha Rush's GPU puzzles https://github.com/srush/GPU-Puzzles
+- raymarch - (WIP) Using the webgpu computation to implement a toy raymarcher
   as an example of non-ML general purpose computation. 
 - webgpu_intro - A minimal from-scratch example of how to use WebGPU directly
   without this library. This is useful to understand the code internals of

examples/gpu_puzzles/run.cpp

@@ -120,7 +120,7 @@ void puzzle3(GPUContext& ctx) {
 // ...
 
 int main(int argc, char **argv) {
-  GPUContext ctx = CreateGPUContext();
+  GPUContext ctx = CreateContext();
   puzzle1(ctx);
   puzzle2(ctx);
   puzzle3(ctx);

examples/hello_world/Makefile

@@ -8,6 +8,10 @@ FASTBUILD_FLAGS = $(FLAGS) -DFASTBUILD:BOOL=ON
 run:
 	mkdir -p build && cd build && cmake .. $(FASTBUILD_FLAGS) && make -j$(NUM_JOBS) $(TARGET) && ./$(TARGET)
 
+watch:
+	@command -v entr >/dev/null 2>&1 || { echo >&2 "Please install entr with 'brew install entr' or 'sudo apt-get install entr'"; exit 1; }
+	mkdir -p build && cd build && cmake .. $(FASTBUILD_FLAGS) && ls ../* ../utils/* | entr -s "rm -f $(TARGET) && make -j$(NUM_JOBS) $(TARGET) && ./$(TARGET)"
+
 clean:
 	read -r -p "This will delete the contents of build/*. Are you sure? [CTRL-C to abort] " response && rm -rf build/*
 

examples/hello_world/run.cpp

@@ -1,5 +1,6 @@
 #include "gpu.h"
 #include "nn/shaders.h"
+#include "utils/logging.h"
 #include <array>
 #include <cstdio>
 
@@ -23,7 +24,8 @@ fn main(
 )";
 
 int main(int argc, char **argv) {
-  GPUContext ctx = CreateGPUContext();
+  log(kDefLog, kInfo, "Hello, gpu.cpp!");
+  GPUContext ctx = CreateContext();
   fprintf(stdout, "\nHello, gpu.cpp\n\n");
   static constexpr size_t N = 3072;
   std::array<float, N> inputArr;
@@ -35,7 +37,7 @@ int main(int argc, char **argv) {
   GPUTensor output = CreateTensor(ctx, {N}, kf32, outputArr.data());
 
   Kernel op = CreateKernel(ctx, CreateShader(kGelu, 256, kf32),
-                           std::array{input}, output);
+                           input, output);
   DispatchKernel(ctx, op);
   Wait(ctx, op.future);
   ToCPU(ctx, output, outputArr.data(), sizeof(outputArr));

examples/raymarch/run.cpp

@@ -76,7 +76,7 @@ int main(int argc, char **argv) {
     float sphereCenterZ;
   } params = {NCOLS, NROWS, 0.5, 0.0, 0.0, 0.0};
 
-  GPUContext ctx = CreateGPUContext();
+  GPUContext ctx = CreateContext();
   GPUTensor devScreen = CreateTensor(ctx, {NROWS, NCOLS}, kf32, screen.data());
   Kernel render = CreateKernel(ctx, ShaderCode{kSDF, 64}, {}, 0, devScreen, params);
   DispatchKernel(ctx, render);