From 667e50498a226103278d0997528cc76979b2c4ef Mon Sep 17 00:00:00 2001
From: Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com>
Date: Wed, 10 Jul 2024 16:19:06 -0400
Subject: Add `float16` support to slang-torch (#4584)

---
 prelude/slang-torch-prelude.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/prelude/slang-torch-prelude.h b/prelude/slang-torch-prelude.h
index 28548e48e..11ffe3b66 100644
--- a/prelude/slang-torch-prelude.h
+++ b/prelude/slang-torch-prelude.h
@@ -107,6 +107,10 @@ TensorView make_tensor_view(torch::Tensor val, const char* name, torch::ScalarTy
         elementSize = 2;
         res.data = (uint8_t*)val.data_ptr<torch::BFloat16>();
         break;
+    case torch::kFloat16:
+        elementSize = 2;
+        res.data = (uint8_t*)val.data_ptr<at::Half>();
+        break;
     case torch::kInt16:
         elementSize = 2;
         res.data = (uint8_t*)val.data_ptr<int16_t>();
-- 
cgit v1.2.3