[Draft] Fix blogpost code

Author: aaronmondal

.bazelignore

@@ -0,0 +1 @@
+.direnv

.gitignore

@@ -1,6 +1,5 @@
 .direnv/
 .pre-commit-config.yaml
-MODULE.bazel.lock
 bazel-*
 compile_commands.json
 lre.bazelrc

finetuning_on_cpu/.bazelrc

@@ -0,0 +1,35 @@
+# Pick up system specific configs.
+common --enable_platform_specific_config
+
+# Don't inherit PATH and LD_LIBRARY_PATH.
+build --incompatible_strict_action_env
+
+# Forbid network access unless explicitly enabled.
+build --sandbox_default_allow_network=false
+
+# Use correct runfile locations.
+build --nolegacy_external_runfiles
+
+# Enable sandboxing for exclusive tests like GPU performance tests.
+test --incompatible_exclusive_test_sandboxed
+
+# Simulate rules_python's initialization with a script and use a statically
+# linked musl Python toolchain.
+common --@rules_python//python/config_settings:bootstrap_impl=script
+common --@rules_python//python/config_settings:py_linux_libc=musl
+# common --@rules_python//python/config_settings:pip_whl_muslc_version=3.13
+
+# Remote optimizations.
+build --remote_download_minimal
+
+# Smaller profiling. Careful. Disabling this might explode remote cache usage.
+build --slim_profile
+build --experimental_profile_include_target_label
+build --noexperimental_profile_include_primary_output
+
+# Use the container image specified in `platforms/BUILD.bazel`.
+build --extra_execution_platforms=@//platforms:python-remote
+build --extra_toolchains=@zig_sdk//toolchain:linux_amd64_musl
+
+# Add user specific flags.
+try-import %workspace%/user.bazelrc

finetuning_on_cpu/.bazelversion

@@ -0,0 +1 @@
+8.2.1

finetuning_on_cpu/.gitignore

@@ -3,6 +3,4 @@ bazel-finetune-repo
 bazel-out
 bazel-bin
 bazel-testlogs
-.bazelrc
 .env
-MODULE.bazel.lock

finetuning_on_cpu/BUILD finetuning_on_cpu/BUILD.bazelfinetuning_on_cpu/BUILD renamed to finetuning_on_cpu/BUILD.bazel

@@ -1,16 +1,22 @@
+bazel_dep(name = "hermetic_cc_toolchain", version = "4.0.0")
+bazel_dep(name = "platforms", version = "0.0.11")
+bazel_dep(name = "rules_cc", version = "0.1.1")
 bazel_dep(name = "rules_python", version = "1.2.0")
 bazel_dep(name = "rules_shell", version = "0.4.1")
 
+# Used internally by rules_python.
+cc_toolchains = use_extension("@hermetic_cc_toolchain//toolchain:ext.bzl", "toolchains")
+use_repo(cc_toolchains, "zig_sdk")
+
+# Python configuration
 python = use_extension("@rules_python//python/extensions:python.bzl", "python")
-python.toolchain(
-    python_version = "3.13",
-)
+python.toolchain(python_version = "3.13")
 
+# Python packages
 pip = use_extension("@rules_python//python/extensions:pip.bzl", "pip")
 pip.parse(
     hub_name = "pypi",
     python_version = "3.13",
     requirements_lock = "//:bazel_requirements_lock.txt",
 )
-
 use_repo(pip, "pypi")

finetuning_on_cpu/MODULE.bazel

@@ -155,3 +155,19 @@ python-dotenv==1.1.0
     # for loading env variables
 scipy==1.15.2
     # for torch to get embeddings
+
+# Manual adjusted below.
+nvidia-cublas-cu12==12.9.0.13
+nvidia-cuda-cupti-cu12==12.9.19
+nvidia-cuda-nvrtc-cu12==12.9.41
+nvidia-cuda-runtime-cu12==12.9.37
+nvidia-cudnn-cu12==9.10.0.56
+nvidia-cufft-cu12==11.4.0.6
+nvidia-curand-cu12==10.3.10.19
+nvidia-cusolver-cu12==11.7.4.40
+nvidia-cusparse-cu12==12.5.9.5
+nvidia-cusparselt-cu12==0.7.1
+nvidia-nccl-cu12==2.26.5
+nvidia-nvjitlink-cu12==12.9.41
+nvidia-nvtx-cu12==12.9.19
+triton==3.3.0

finetuning_on_cpu/MODULE.bazel.lock

@@ -0,0 +1,12 @@
+platform(
+    name = "python-remote",
+    constraint_values = [
+        "@platforms//os:linux",
+        "@platforms//cpu:x86_64",
+    ],
+    exec_properties = {
+        "OSFamily": "Linux",
+        "ISA": "x86-64",
+        "container-image": "docker.io/evaanahmed2001/python-bazel-env@sha256:8de13199d587964b218c0b671272b42031cf4944b2f426e6eee7d7542802bf7c",
+    },
+)