lym00 commited on
Commit
409dba5
·
verified ·
1 Parent(s): 4a2a748

Update test_sageattn3.py

Browse files
Files changed (1) hide show
  1. test_sageattn3.py +1 -1
test_sageattn3.py CHANGED
@@ -44,7 +44,7 @@ def main():
44
 
45
  out_sage = sageattn_blackwell(q, k, v, is_causal=False)
46
  print("sage vs math:", get_rtol_atol(out_sage, out_math))
47
- print("The above (except max_rtol) should be < 0.05 (on RTX 20xx/30xx) or < 0.1 (on RTX 40xx/50xx)")
48
 
49
 
50
  if __name__ == "__main__":
 
44
 
45
  out_sage = sageattn_blackwell(q, k, v, is_causal=False)
46
  print("sage vs math:", get_rtol_atol(out_sage, out_math))
47
+ print("The above (except max_rtol) should be < 0.1 (on RTX 50xx)")
48
 
49
 
50
  if __name__ == "__main__":