Update quant.py

quant.py

@@ -657,7 +657,8 @@ __global__ void VecQuant8MatMulKernel(
   atomicAdd(&mul[b * width + w], res);
 }
     '''
-    open("quant_cuda_kernel.cu","w").write(cucode).close()
+    open("quant_cuda_kernel.cu","w") as f:
+        f.write(cucode)
     cppcode = '''
 #include <torch/all.h>
 #include <torch/python.h>
@@ -730,7 +731,8 @@ PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
   m.def("vecquant8matmul", &vecquant8matmul, "Vector 8-bit Quantized Matrix Multiplication (CUDA)");
 }
     '''
-    open("quant_cuda.cpp","w").write(cppcode).close()
+    with open("quant_cuda.cpp","w") as f:
+        f.write(cppcode)
     setup(
         name='quant_cuda',
         ext_modules=[cpp_extension.CUDAExtension(