Update fmha_utils.h

bug fix
ROCm · Jun 15, 2023 · 8559ccd · 8559ccd
1 parent ee0665c
commit 8559ccd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/csrc/flash_attn_rocm/src/fmha_utils.h b/csrc/flash_attn_rocm/src/fmha_utils.h
@@ -96,7 +96,7 @@ static inline size_t get_size_in_bytes( size_t n, DataType dtype ) {
 
 static std::tuple<uint64_t, uint64_t> unpack(at::PhiloxCudaState arg) {
   if (arg.captured_) {
-    return std::make_tuple(static_cast<int64_t>(*arg.seed_.ptr), static_cast<uint64_t>(*(arg.offset_.ptr) + arg.offset_intragraph_));
+    return std::make_tuple(static_cast<uint64_t>(*arg.seed_.ptr), static_cast<uint64_t>(*(arg.offset_.ptr) + arg.offset_intragraph_));
   } else {
     return std::make_tuple(arg.seed_.val, arg.offset_.val);
   }