Add custom Vite plugin to strip 'module' attribute from script tags (#11)

Summary:
- Introduced a new Vite plugin named 'strip-module-attr' that transforms script tags by replacing the 'type="module"' attribute with 'defer' to fix unidentified module type internally

Pull Request resolved: #11

Reviewed By: xuzhao9

Differential Revision: D76308714

Pulled By: FindHao

fbshipit-source-id: 93107d1a162126c55d9c9d159fa167031d573aae

Author: FindHao

README.md

@@ -100,20 +100,21 @@ First, integrate TritonParse with your Triton/PyTorch code to generate trace fil
 
 ```python
 import torch
+# === TritonParse ===
 import tritonparse.structured_logging
-
 # Initialize structured logging to capture Triton compilation events
+# This will generate NDJSON trace logs in ./logs/
 tritonparse.structured_logging.init("./logs/")
+# === TritonParse end ===
 
+# The below is your original Triton/PyTorch 2 code
 # Example: Using with torch.compile
 def your_kernel():
     # Your PyTorch/Triton kernel code
     pass
 
 compiled_kernel = torch.compile(your_kernel)
-result = compiled_kernel()  # This will generate NDJSON trace logs in ./logs/
-
-# The trace files can then be analyzed using the web interface
+result = compiled_kernel()
 ```
 
 ### 2. Analyze with Web Interface

website/vite.config.ts

@@ -9,7 +9,17 @@ const packageJson = JSON.parse(
 
 export default defineConfig({
   plugins: [
-    react()
+    react(),
+    {
+      name: 'strip-module-attr',
+      enforce: 'post',
+      transformIndexHtml(html) {
+        return html.replace(
+          /<script\s+type=["']module["']\s+([^>]*?)src=/g,
+          '<script defer $1src='
+        )
+      }
+    }
   ],
 
   base: './',