Full rewrite

.gitignore

@@ -56,3 +56,6 @@ imgui.ini
 *.npz
 /cmake-build-*
 *.pyc
+/.cmake
+/CMakeFiles
+/.debug

.gitmodules

@@ -4,9 +4,9 @@
 [submodule "external/glfw"]
 	path = external/glfw
 	url = https://github.com/glfw/glfw.git
-[submodule "external/glad"]
-	path = external/glad
-	url = https://github.com/Dav1dde/glad.git
 [submodule "external/imgui"]
 	path = external/imgui
 	url = https://github.com/ocornut/imgui.git
+[submodule "Python/external/shaderc"]
+	path = Python/external/shaderc
+	url = https://github.com/google/shaderc

.run/TensorFrost.run.xml

@@ -1,5 +1,5 @@
 <component name="ProjectRunConfigurationManager">
-  <configuration default="false" name="TensorFrost" type="CMakeRunConfiguration" factoryName="Application" PROGRAM_PARAMS="H:\TensorFrost\examples\debug.py" REDIRECT_INPUT="false" ELEVATE="false" USE_EXTERNAL_CONSOLE="false" EMULATE_TERMINAL="false" WORKING_DIR="file://$PROJECT_DIR$" PASS_PARENT_ENVS_2="true" PROJECT_NAME="TensorFrost" TARGET_NAME="TensorFrost" CONFIG_NAME="Debug-Visual Studio" RUN_PATH="$USER_HOME$/AppData/Local/Programs/Python/Python312/python.exe">
+  <configuration default="false" name="TensorFrost" type="CMakeRunConfiguration" factoryName="Application" PROGRAM_PARAMS="H:\TensorFrost\examples\Slang\mandelbrot.py" REDIRECT_INPUT="false" ELEVATE="false" USE_EXTERNAL_CONSOLE="false" EMULATE_TERMINAL="false" WORKING_DIR="file://$PROJECT_DIR$" PASS_PARENT_ENVS_2="true" PROJECT_NAME="TensorFrost" TARGET_NAME="TensorFrost" CONFIG_NAME="RelWithDebInfo" RUN_PATH="C:/Python313/python.exe">
     <method v="2">
       <option name="com.jetbrains.cidr.execution.CidrBuildBeforeRunTaskProvider$BuildBeforeRunTask" enabled="true" />
     </method>

AGENTS.md

@@ -0,0 +1,10 @@
+# Agent Guide
+
+Follow these expectations whenever you work in this repository:
+
+1. **Full rebuild & virtual environment** — Run `setup_python_env.cmd` from the repo root. It configures the Python virtual environment and performs a clean rebuild so you start from a consistent state.
+2. **Partial rebuilds** — Use CMake for incremental builds. Invoke the appropriate CMake build command (for example, `cmake --build <build-directory> --target <target>`) to rebuild only what you need.
+3. **C++ changes** — Any edits under `TensorFrost/` or other C++ sources require a rebuild before the changes take effect.
+4. **Python script changes** - After edits of python script, you should run them to make sure they work correctly. No recompilation needed.
+5. **API validation** — After modifying functionality, run the relevant tests in the `tests/` folder to confirm the Python API still behaves as expected.
+6. **Scenario validation** — Run the sample programs in the `examples/` folder to make sure the updated stack handles more complex end-to-end flows.

CMakeLists.txt

@@ -34,10 +34,11 @@ set(GLFW_BUILD_TESTS OFF CACHE BOOL "" FORCE)
 set(GLFW_BUILD_EXAMPLES OFF CACHE BOOL "" FORCE)
 set(PYBIND11_FINDPYTHON ON)
 
+find_package(Vulkan REQUIRED)
+
 add_subdirectory(external/pybind11)
 add_subdirectory(external/glfw)
-add_subdirectory(external/glad/cmake)
 add_subdirectory(TensorFrost)
 add_subdirectory(examples)
 
-set_property(DIRECTORY ${CMAKE_SOURCE_DIR} PROPERTY VS_STARTUP_PROJECT TensorFrost)
+set_property(DIRECTORY ${CMAKE_SOURCE_DIR} PROPERTY VS_STARTUP_PROJECT TensorFrost)

ProtoIR.txt

@@ -0,0 +1,16 @@
+Node = [name, arguments, attributes] 
+
+
+n = input_dim(attributes{type=int32, input_index=0, dim_index=0})
+a = input(args{shape=[n]}, attributes{type=float32, input_index=0})
+b = sin(args{input=[a], shape=[n]}, attributes{type=float32})
+c = load(args{input=[b], indices=[0]}, attributes{type=float32})
+d = average(args{input=[b]}, attributes{type=float32})
+res = div(args{input=[d,c]}, attributes{type=float32})
+ids = parallel(args{shape=[n, n]}, attributes{type=tuple}) {
+    i = unpack_tuple(args{input=[ids,0]}, attributes{type=int32})
+    j = unpack_tuple(args{input=[ids,0]}, attributes{type=int32})
+    a0 = load(args{input=[b], indices=[i]}, attributes{type=float32})
+    a1 = load(args{input=[b], indices=[j]}, attributes{type=float32})
+    outer = mul(args{input=[a0, a1]}, attributes{type=float32, output_index=0})
+}

Python/TensorFrost/__init__.py

@@ -4,7 +4,6 @@
 from . import regularizers
 from . import clipping
 from . import random
-from . import sort
 from .default import *
 
 # def compile(func):

Python/TensorFrost/clipping.py

@@ -1,5 +1,5 @@
-from .optimizers import *
-
-clamp = ModuleOptimizer.ClippingType.Clamp
-norm = ModuleOptimizer.ClippingType.Norm
-none = ModuleOptimizer.ClippingType.None_
+# from .optimizers import *
+#
+# clamp = ModuleOptimizer.ClippingType.Clamp
+# norm = ModuleOptimizer.ClippingType.Norm
+# none = ModuleOptimizer.ClippingType.None_

Python/TensorFrost/default.py

@@ -1,14 +1,14 @@
-from . import TensorFrost as tf
-
-def zeros_like(tensor):
-    return tf.zeros(tensor.shape, tensor.type)
-
-def eye(n):
-    i, j = tf.indices([n, n])
-    return tf.select(i == j, 1.0, 0.0)
-
-def eye_like(tensor):
-    return eye(tensor.shape[0])
-
-def ones_like(tensor):
-    return tf.ones(tensor.shape, tensor.type)
+# from . import TensorFrost as tf
+#
+# def zeros_like(tensor):
+#     return tf.zeros(tensor.shape, tensor.type)
+#
+# def eye(n):
+#     i, j = tf.indices([n, n])
+#     return tf.select(i == j, 1.0, 0.0)
+#
+# def eye_like(tensor):
+#     return eye(tensor.shape[0])
+#
+# def ones_like(tensor):
+#     return tf.ones(tensor.shape, tensor.type)