Update visibility checks in Vulkan to support geom shaders

Bug: skia:
Change-Id: Ie80bea38e644647e5a09e95053ddace30521361d
Reviewed-on: https://skia-review.googlesource.com/23300
Reviewed-by: Chris Dalton <csmartdalton@google.com>
Commit-Queue: Greg Daniel <egdaniel@google.com>
diff --git a/src/gpu/vk/GrVkPipelineStateDataManager.cpp b/src/gpu/vk/GrVkPipelineStateDataManager.cpp
index 5d96bd7..ee0b6a7 100644
--- a/src/gpu/vk/GrVkPipelineStateDataManager.cpp
+++ b/src/gpu/vk/GrVkPipelineStateDataManager.cpp
@@ -33,8 +33,7 @@
             uniform.fType = uniformInfo.fVariable.getType();
         );
 
-        if (kVertex_GrShaderFlag == uniformInfo.fVisibility ||
-            kGeometry_GrShaderFlag == uniformInfo.fVisibility) {
+        if (!(kFragment_GrShaderFlag & uniformInfo.fVisibility)) {
             uniform.fBinding = GrVkUniformHandler::kGeometryBinding;
         } else {
             SkASSERT(kFragment_GrShaderFlag == uniformInfo.fVisibility);