Manually delete tempfile after it is not needed any more

This works on Linux and Windows and on all versions of python.

Signed-off-by: Gregory Shimansky <[email protected]>

Author: gshimansky

python/test/unit/language/test_core.py

@@ -2602,8 +2602,9 @@ def test_scan_layouts(M, N, src_layout, axis, device):
 
     with tempfile.NamedTemporaryFile(mode='w', suffix='.ttgir', delete=False) as f:
         f.write(ir)
-        f.flush()
+        f.close()
         kernel = triton.compile(f.name)
+        os.remove(f.name)
     rs = RandomState(17)
     x = rs.randint(-100, 100, (M, N)).astype('int32')
 
@@ -2736,9 +2737,9 @@ def test_reduce_layouts(M, N, src_layout, axis, epilogue_kind, dtype_str, reduce
 
     with tempfile.NamedTemporaryFile(mode='w', suffix='.ttgir', delete=False) as f:
         f.write(ir)
-        f.flush()
         f.close()
         kernel = triton.compile(f.name)
+        os.remove(f.name)
 
     rs = RandomState(17)
     x = numpy_random((M, N), dtype_str=dtype_str, rs=rs, low=0, high=10)
@@ -2791,8 +2792,9 @@ def test_store_op(M, src_layout, device):
 
     with tempfile.NamedTemporaryFile(mode='w', suffix='.ttgir', delete=False) as f:
         f.write(ir)
-        f.flush()
+        f.close()
         store_kernel = triton.compile(f.name)
+        os.remove(f.name)
 
     rs = RandomState(17)
     x = rs.randint(0, 4, (M, 1)).astype('float32')
@@ -2841,8 +2843,9 @@ def test_convert1d(M, src_layout, dst_layout, src_dim, dst_dim, device):
     """
     with tempfile.NamedTemporaryFile(mode='w', suffix='.ttgir', delete=False) as f:
         f.write(ir)
-        f.flush()
+        f.close()
         kernel = triton.compile(f.name)
+        os.remove(f.name)
 
     rs = RandomState(17)
     x = rs.randint(0, 4, (M, )).astype('int32')
@@ -2923,8 +2926,9 @@ def test_chain_reduce(M, N, src_layout, op, device, first_axis):
     """
     with tempfile.NamedTemporaryFile(mode='w', suffix='.ttgir', delete=False) as f:
         f.write(ir)
-        f.flush()
+        f.close()
         kernel = triton.compile(f.name)
+        os.remove(f.name)
 
     rs = RandomState(17)
     x = rs.randint(0, 4, (M, N)).astype('int32')
@@ -5300,8 +5304,9 @@ def test_convert2d(M, N, src_layout, interm_layout, dst_layout, dtype, device):
 
     with tempfile.NamedTemporaryFile(mode='w', suffix='.ttgir', delete=False) as f:
         f.write(ir)
-        f.flush()
+        f.close()
         kernel = triton.compile(f.name)
+        os.remove(f.name)
     kernel[(1, 1, 1)](x.data_ptr(), z.data_ptr())
 
     assert torch.equal(z, x)
@@ -5413,8 +5418,9 @@ def do_test(src_layout, dst_layout):
 
         with tempfile.NamedTemporaryFile(mode='w', suffix='.ttgir', delete=False) as f:
             f.write(ir)
-            f.flush()
+            f.close()
             kernel = triton.compile(f.name)
+            os.remove(f.name)
         kernel[(1, 1, 1)](x.data_ptr(), z.data_ptr())
 
         assert torch.equal(z, x)