triton-windows 3.3.0a0.post15__cp312-cp312-win_amd64.whl → 3.3.0a0.post17__cp312-cp312-win_amd64.whl

triton/backends/nvidia/compiler.py

@@ -403,11 +403,10 @@ class CUDABackend(BaseBackend):
 
     def make_cubin(self, src, metadata, opt, capability):
         ptxas, _ = get_ptxas(self.target.arch)
-        # On Windows, we need to set delete=False, close the temp file before reading it, and manually remove it
         with tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.ptx') as fsrc, \
             tempfile.NamedTemporaryFile(delete=False, mode='r', suffix='.log') as flog:
             fsrc.write(src)
-            fsrc.close()
+            fsrc.flush()
             fbin = fsrc.name + '.o'
 
             line_info = ["-lineinfo", "-suppress-debug-info"] if os.environ.get("TRITON_DISABLE_LINE_INFO",
@@ -417,10 +416,8 @@ class CUDABackend(BaseBackend):
             opt_level = ['--opt-level', '0'] if os.environ.get("DISABLE_PTXAS_OPT", "0") == "1" else []
             ptxas_cmd = [ptxas, *line_info, *fmad, '-v', *opt_level, f'--gpu-name={arch}', fsrc.name, '-o', fbin]
             try:
-                subprocess.run(ptxas_cmd, check=True, close_fds=False, stdout=flog, stderr=flog)
-                flog.close()
+                subprocess.run(ptxas_cmd, check=True, close_fds=True, stdout=flog, stderr=flog)
             except subprocess.CalledProcessError as e:
-                flog.close()
                 with open(flog.name) as log_file:
                     log = log_file.read()
 
@@ -434,13 +431,14 @@ class CUDABackend(BaseBackend):
                 raise PTXASError(f"{error}\n"
                                  f"`ptxas` stderr:\n{log}\n"
                                  f'Repro command: {" ".join(ptxas_cmd)}\n')
-            finally:
-                try_remove(fsrc.name)
-                try_remove(flog.name)
 
-            with open(fbin, 'rb') as f:
-                cubin = f.read()
-            try_remove(fbin)
+        with open(fbin, 'rb') as f:
+            cubin = f.read()
+        try_remove(fbin)
+
+        # It's better to remove the temp files outside the context managers
+        try_remove(fsrc.name)
+        try_remove(flog.name)
         return cubin
 
     def add_stages(self, stages, options):

triton/runtime/cache.py

@@ -131,7 +131,15 @@ class FileCacheManager(CacheManager):
             f.write(data)
         # Replace is guaranteed to be atomic on POSIX systems if it succeeds
         # so filepath cannot see a partial write
-        os.replace(temp_path, filepath)
+        try:
+            os.replace(temp_path, filepath)
+        except PermissionError:
+            # Ignore PermissionError on Windows because it happens when another process already
+            # put a file into the cache and locked it by opening it.
+            if os.name == "nt":
+                os.remove(temp_path)
+            else:
+                raise
         os.removedirs(temp_dir)
         return filepath
 

{triton_windows-3.3.0a0.post15.dist-info → triton_windows-3.3.0a0.post17.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: triton-windows
-Version: 3.3.0a0.post15
+Version: 3.3.0a0.post17
 Summary: A language and compiler for custom Deep Learning operations
 Home-page: https://github.com/woct0rdho/triton-windows
 Author: Philippe Tillet, Dian Wu

{triton_windows-3.3.0a0.post15.dist-info → triton_windows-3.3.0a0.post17.dist-info}/RECORD

@@ -4,7 +4,7 @@ triton/_utils.py,sha256=5RiCLwW14w0Q3mdZ-9yz-VO5KiSexNj9xeDt4gaNsvE,1014
 triton/errors.py,sha256=8WfnuRKLG578mgY6cBA3ECruVMf9ULEKFNgRcJ6IhWM,89
 triton/testing.py,sha256=ivFf1Fq9frmfVahaVUp0bgJxmvVZNACZfj3Sai6zfAs,20048
 triton/windows_utils.py,sha256=aQMItmuZNXaki8zSB7HMvhy1RsXlmz1GcO-lORm1IIk,10852
-triton/_C/libtriton.pyd,sha256=aOtQ6tmASJ_qFMd6IPQX0Zbmn1ByGbHM6BXF2r8Murc,86836224
+triton/_C/libtriton.pyd,sha256=Ly4JelI_9qSPSKD4rraH6fY20GdrCU0Ot9L0NPupJjk,86836224
 triton/backends/__init__.py,sha256=opAo_vgEMt3tLO_bYFrYGksnIu0qohbmyuu_s3-rNAs,1595
 triton/backends/compiler.py,sha256=ymaG0kpveAuESbQ9QZ0RyXjr0Aq4el_G5XGYogJ2gNA,3588
 triton/backends/driver.py,sha256=AN60upJlPgia0JwvZ8vIVgLITNPuI0fdz8zMIIHPpF4,1450
@@ -112,7 +112,7 @@ triton/backends/amd/lib/asanrtl.bc,sha256=1xv2RlU3WvbdsghHlmhwiHewGM2B5dKts5bERM
 triton/backends/amd/lib/ockl.bc,sha256=wQKCzkKukIHbu0lyjKUYlhndc7S27xto6L54J0Bn-C0,246124
 triton/backends/amd/lib/ocml.bc,sha256=UPNTXW0gCXUNB-c6orSYwb-mz9_mjUc7zny_vfFza44,205964
 triton/backends/nvidia/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-triton/backends/nvidia/compiler.py,sha256=R0WKIM_JsMcd3nlwnrlB8pXPjyDj6egaaFxNwi2hneM,18588
+triton/backends/nvidia/compiler.py,sha256=iHbIl-HXSgD3S5eJIXbQhr-Cyk9Gq_g8RXDFZxAGUF4,18444
 triton/backends/nvidia/driver.c,sha256=RWzdn9_RWaaVxXKtdrvxoRg4yR7WSH7ELRyjGDYKwBM,18880
 triton/backends/nvidia/driver.py,sha256=BcEzr0_ONM8H_DKoCbNuO1jRpBM_XHKBgt_Y2bG9aZw,20902
 triton/backends/nvidia/bin/ptxas.exe,sha256=iCva9hIYg-Q2NybchwaQJFkwDzNiliFOTDdZrHPLb6A,24732160
@@ -143,7 +143,7 @@ triton/runtime/__init__.py,sha256=mKL5cqIBDUw2WO80NRCh4s1G8KYaqgM59TTAbTkPPjQ,62
 triton/runtime/_allocation.py,sha256=zaW4B7I7c-2rkVuN7IZaUB6IQSI1t4FvnTPZH-r7DTk,798
 triton/runtime/autotuner.py,sha256=h1I6TDPPYtMvuwOkeprcSD3VKKGXk952jXN64oPIFXs,16844
 triton/runtime/build.py,sha256=6Ie4gn4-Zk1DIW3o-Nd002IhtEm_yikXh9o6tUm6iao,3279
-triton/runtime/cache.py,sha256=Vs9l4DeRhKXyB-hfDBrp5v85OO6vkkXTHTVjZ7zu0xg,10260
+triton/runtime/cache.py,sha256=uoU1UH3HPxkxT0r-69HIZgLBls3T-TDxJedRULX9lbM,10583
 triton/runtime/driver.py,sha256=VZ-883Xri71R72lHB6usIpLo3gGLbZJkAlLP3ewWSpc,1509
 triton/runtime/errors.py,sha256=CwfJXciwel_-K3BfQfKUpLPDWrSyTnGsfJkqJojrdfQ,1052
 triton/runtime/interpreter.py,sha256=RVzNGSXyLlg14LgjJ5BZ3qbqfK49FccG-pXaIXXXE2g,59331
@@ -247,7 +247,7 @@ triton/tools/link.py,sha256=u7qtfZRLriZkAMEGNvj8YF-k1cthmLL7BwHYqBgT63E,11871
 triton/tools/mxfp.py,sha256=YQdpBrGkOVNOtnLeRjMCeVFHWkSwUubGeWsItIjO8TU,11737
 triton/tools/extra/cuda/compile.c,sha256=Me7beHPc6WNTwjg85H84DUMCRu4KJdVK2hNNgvlhBZ4,2126
 triton/tools/extra/cuda/compile.h,sha256=n9QKIFZTL4RSsiXtAxBP9XGSnxjyaevQQ9bBpwDsvAg,332
-triton_windows-3.3.0a0.post15.dist-info/METADATA,sha256=pnp3W_MNnmJ3LJy6cNWAuAgOdrpJo3AfJvfERYBNwB4,1629
-triton_windows-3.3.0a0.post15.dist-info/WHEEL,sha256=A8mRFNvJcDL8dRPZ6k2ICKEOXwE8pzYFXYxEla0rR0g,101
-triton_windows-3.3.0a0.post15.dist-info/top_level.txt,sha256=KhMzHYsArnZ3IkjAQ-xLnx1n_FjvEpJNelg2xPiDl-U,254
-triton_windows-3.3.0a0.post15.dist-info/RECORD,,
+triton_windows-3.3.0a0.post17.dist-info/METADATA,sha256=Hj5t_K8_B7fKw2Tp4ki_-rOSVDJ4TWFlJMJ8S_tqShc,1629
+triton_windows-3.3.0a0.post17.dist-info/WHEEL,sha256=9_mdKl-_Yr--f6EOmQ6F9FLhZL1o0okl1kr9b_edmVI,101
+triton_windows-3.3.0a0.post17.dist-info/top_level.txt,sha256=KhMzHYsArnZ3IkjAQ-xLnx1n_FjvEpJNelg2xPiDl-U,254
+triton_windows-3.3.0a0.post17.dist-info/RECORD,,

{triton_windows-3.3.0a0.post15.dist-info → triton_windows-3.3.0a0.post17.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (76.0.0)
+Generator: setuptools (77.0.1)
 Root-Is-Purelib: false
 Tag: cp312-cp312-win_amd64