From 26a7cf79526b86a3dff4084d42dde8f1a8c9ac1d Mon Sep 17 00:00:00 2001 From: Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com> Date: Fri, 8 Sep 2023 17:03:14 -0400 Subject: Remove unsupported torch types + add bool type. (#3197) Co-authored-by: Yong He --- prelude/slang-torch-prelude.h | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'prelude') diff --git a/prelude/slang-torch-prelude.h b/prelude/slang-torch-prelude.h index a10e0070f..dee116261 100644 --- a/prelude/slang-torch-prelude.h +++ b/prelude/slang-torch-prelude.h @@ -123,6 +123,10 @@ TensorView make_tensor_view(torch::Tensor val, const char* name, torch::ScalarTy elementSize = 8; res.data = (uint8_t*)val.data_ptr(); break; + case torch::kBool: + elementSize = 1; + res.data = (uint8_t*)val.data_ptr(); + break; } if (val.dim() > kSlangTorchTensorMaxDim) -- cgit v1.2.3