Skip to content

Commit a9cbb38

Browse files
jorgep31415facebook-github-bot
authored andcommitted
Enable uint8 dtype in shaders (#5932)
Summary: Pull Request resolved: #5932 TSIA ghstack-source-id: 246607061 exported-using-ghexport bypass-github-export-checks Reviewed By: SS-JIA Differential Revision: D63918660 fbshipit-source-id: 3034b31899e8079ccb1d443bade4a0185997bb7a
1 parent 0a11e99 commit a9cbb38

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)