From f5fb13ee143a8d8de569db50c1cd57983d28949d Mon Sep 17 00:00:00 2001
From: Shivam7-1 <55046031+Shivam7-1@users.noreply.github.com>
Date: Tue, 31 Dec 2024 16:10:31 +0530
Subject: [PATCH] Created fuzz_target.py

---
 fuzz/fuzz_target.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 fuzz/fuzz_target.py

diff --git a/fuzz/fuzz_target.py b/fuzz/fuzz_target.py
new file mode 100644
index 0000000..92a909a
--- /dev/null
+++ b/fuzz/fuzz_target.py
@@ -0,0 +1,28 @@
+import atheris
+import sys
+import os
+import json
+import torch
+from safetensors.torch import load_file, save_file
+from kernel import weight_dequant
+
+def TestOneInput(data):
+    fdp = atheris.FuzzedDataProvider(data)
+    try:
+        fp8_path = fdp.ConsumeUnicodeNoSurrogates(50)
+        bf16_path = fdp.ConsumeUnicodeNoSurrogates(50)
+        torch.set_default_dtype(torch.bfloat16)
+        os.makedirs(bf16_path, exist_ok=True)
+        model_index_file = os.path.join(fp8_path, "model.safetensors.index.json")
+        with open(model_index_file, "r") as f:
+            model_index = json.load(f)
+        weight_map = model_index["weight_map"]
+        for tensor_name in weight_map:
+            file_name = weight_map[tensor_name]
+            file_path = os.path.join(fp8_path, file_name)
+            load_file(file_path, device="cuda")
+    except Exception as e:
+        pass
+
+atheris.Setup(sys.argv, TestOneInput)
+atheris.Fuzz()