Skip to content

Commit 71ba888

Browse files
committed
[5/n][ET-VK] Enable uint8 dtype in shaders
TSIA Differential Revision: [D63918660](https://our.internmc.facebook.com/intern/diff/D63918660/) ghstack-source-id: 246607061 Pull Request resolved: #5932
1 parent 84f5a56 commit 71ba888

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

backends/vulkan/runtime/graph/ops/utils/ShaderNameUtils.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,10 @@ void add_dtype_suffix(std::string& kernel_name, const vkapi::ScalarType dtype) {
4747
case vkapi::kQInt8:
4848
kernel_name += "_int8";
4949
break;
50+
case vkapi::kByte:
51+
case vkapi::kQUInt8:
52+
kernel_name += "_uint8";
53+
break;
5054
default:
5155
break;
5256
}

0 commit comments

Comments
 (0)