update builds

build/torch27-cxx11-cu126-x86_64-linux/sage_attention/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _sage_attention_44b112f_dirty
-ops = torch.ops._sage_attention_44b112f_dirty
+from . import _sage_attention_af2d0c0_dirty
+ops = torch.ops._sage_attention_af2d0c0_dirty
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_sage_attention_44b112f_dirty::{op_name}"
+    return f"_sage_attention_af2d0c0_dirty::{op_name}"

build/torch27-cxx11-cu126-x86_64-linux/sage_attention/{_sage_attention_44b112f_dirty.abi3.so → _sage_attention_af2d0c0_dirty.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b577da1986b76b2571e8dd55412621e6fc85fe1a2f847bc0a5af9851bf388cf2
+oid sha256:83f3b3d1c1371cf577a4e2c2fa3bbeef137aa93a89cf380816c14e650b1449f6
 size 26037568

build/torch27-cxx11-cu128-x86_64-linux/sage_attention/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _sage_attention_44b112f_dirty
-ops = torch.ops._sage_attention_44b112f_dirty
+from . import _sage_attention_af2d0c0_dirty
+ops = torch.ops._sage_attention_af2d0c0_dirty
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_sage_attention_44b112f_dirty::{op_name}"
+    return f"_sage_attention_af2d0c0_dirty::{op_name}"

build/torch27-cxx11-cu128-x86_64-linux/sage_attention/{_sage_attention_44b112f_dirty.abi3.so → _sage_attention_af2d0c0_dirty.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d47c952dd9781283ff0dcbd533779de33b0bfa1966dcc0cc8accd0412217c1c5
+oid sha256:871d2abf021f7175f2a66cd9f3599fdd88c9be0c98df1bb4d09f9905d955405f
 size 26553840

build/torch28-cxx11-cu126-x86_64-linux/sage_attention/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _sage_attention_44b112f_dirty
-ops = torch.ops._sage_attention_44b112f_dirty
+from . import _sage_attention_af2d0c0_dirty
+ops = torch.ops._sage_attention_af2d0c0_dirty
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_sage_attention_44b112f_dirty::{op_name}"
+    return f"_sage_attention_af2d0c0_dirty::{op_name}"

build/torch28-cxx11-cu126-x86_64-linux/sage_attention/{_sage_attention_44b112f_dirty.abi3.so → _sage_attention_af2d0c0_dirty.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:28e181de0c6388653fb4b8b2d7347f1f547fc84fe7dc45bc66db9b1431d141bc
+oid sha256:26b18ae63bccd4c5926533ffa1d0995e7bf3faf7919c0c55e1b829267ac73afd
 size 26037392

build/torch28-cxx11-cu128-x86_64-linux/sage_attention/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _sage_attention_44b112f_dirty
-ops = torch.ops._sage_attention_44b112f_dirty
+from . import _sage_attention_af2d0c0_dirty
+ops = torch.ops._sage_attention_af2d0c0_dirty
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_sage_attention_44b112f_dirty::{op_name}"
+    return f"_sage_attention_af2d0c0_dirty::{op_name}"

build/torch28-cxx11-cu128-x86_64-linux/sage_attention/{_sage_attention_44b112f_dirty.abi3.so → _sage_attention_af2d0c0_dirty.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:826ab66e6c33b3b2b17c30371934a55e972d560197c5492f4dedf6fcc29f1a1e
-size 26553920
+oid sha256:2681241cb3fee535e10ba52179293982bca60a5fed972404fdec8ae5fa848175
+size 26549824

build/torch28-cxx11-cu129-x86_64-linux/sage_attention/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _sage_attention_44b112f_dirty
-ops = torch.ops._sage_attention_44b112f_dirty
+from . import _sage_attention_af2d0c0_dirty
+ops = torch.ops._sage_attention_af2d0c0_dirty
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_sage_attention_44b112f_dirty::{op_name}"
+    return f"_sage_attention_af2d0c0_dirty::{op_name}"

build/torch28-cxx11-cu129-x86_64-linux/sage_attention/_sage_attention_44b112f_dirty.abi3.so

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:824faeacc05dc7d676acaa9a005d5f4d7e62f47c361eb58a085f020e21fde29e
-size 26612144

build/torch28-cxx11-cu129-x86_64-linux/sage_attention/_sage_attention_af2d0c0_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ff47cafcc3abed4dc02589ee11c315f3b88f65a0510caa89a07825ccd8ea1a48
+size 26608048

torch-ext/torch_binding.h

@@ -226,10 +226,7 @@ void mean_scale_fuse_quant_cuda(
 void sm_check_89(torch::Tensor x, std::string op_name) {
     int device_index = x.get_device();
     const auto& prop = at::cuda::getDeviceProperties(device_index);
-
-    std::cerr << "sm_check_89: prop->major: " << prop->major << std::endl;
-    std::cerr << "sm_check_89: prop->minor: " << prop->minor << std::endl;
-
+  
     if (prop->major < 8 || (prop->major == 8 && prop->minor < 9)) {
         TORCH_CHECK(false, op_name + " requires compute capability 8.9+");
     }
@@ -239,9 +236,6 @@ void sm_check_90(torch::Tensor x, std::string op_name) {
     int device_index = x.get_device();
     const auto& prop = at::cuda::getDeviceProperties(device_index);
 
-    std::cerr << "sm_check_90: prop->major: " << prop->major << std::endl;
-    std::cerr << "sm_check_90: prop->minor: " << prop->minor << std::endl;
-
     if (prop->major < 9) {
         TORCH_CHECK(false, op_name + " requires compute capability 9.0+");
     }
@@ -250,8 +244,7 @@ void sm_check_90(torch::Tensor x, std::string op_name) {
 void sm_check_80(torch::Tensor x, std::string op_name) {
     int device_index = x.get_device();
     const auto& prop = at::cuda::getDeviceProperties(device_index);
-    std::cerr << "sm_check_80: prop->major: " << prop->major << std::endl;
-    std::cerr << "sm_check_80: prop->minor: " << prop->minor << std::endl;
+    
     if (prop->major < 8) {
         TORCH_CHECK(false, op_name + " requires compute capability 8.0+");
     }