tu: Fix vk2tu_*_stage flag type

New flags were silently getting dropped.

Fixes: 59259a01671 ("tu: Convert to sync2 entrypoints")
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/24162>

diff --git a/src/freedreno/vulkan/tu_cmd_buffer.cc b/src/freedreno/vulkan/tu_cmd_buffer.cc
index 1b74e91..e37495c 100644 (file)
--- a/src/freedreno/vulkan/tu_cmd_buffer.cc
+++ b/src/freedreno/vulkan/tu_cmd_buffer.cc
@@ -3256,10 +3256,10 @@ vk2tu_single_stage(VkPipelineStageFlags2 vk_stage, bool dst)
 }
 
 static enum tu_stage
-vk2tu_src_stage(VkPipelineStageFlags vk_stages)
+vk2tu_src_stage(VkPipelineStageFlags2 vk_stages)
 {
    enum tu_stage stage = TU_STAGE_CP;
-   u_foreach_bit (bit, vk_stages) {
+   u_foreach_bit64 (bit, vk_stages) {
       enum tu_stage new_stage = vk2tu_single_stage(1ull << bit, false);
       stage = MAX2(stage, new_stage);
    }
@@ -3268,10 +3268,10 @@ vk2tu_src_stage(VkPipelineStageFlags vk_stages)
 }
 
 static enum tu_stage
-vk2tu_dst_stage(VkPipelineStageFlags vk_stages)
+vk2tu_dst_stage(VkPipelineStageFlags2 vk_stages)
 {
    enum tu_stage stage = TU_STAGE_PS;
-   u_foreach_bit (bit, vk_stages) {
+   u_foreach_bit64 (bit, vk_stages) {
       enum tu_stage new_stage = vk2tu_single_stage(1ull << bit, true);
       stage = MIN2(stage, new_stage);
    }