From ef54ae93fc161a25209dea3eca061b9482ae4fa9 Mon Sep 17 00:00:00 2001
From: Chin-Yun Yu <cy006@bluck.eecs.qmul.ac.uk>
Date: Mon, 25 Mar 2024 16:00:14 +0000
Subject: [PATCH] refactor

---
 torchcomp/core.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/torchcomp/core.py b/torchcomp/core.py
index 7e4c6de..fe203f1 100644
--- a/torchcomp/core.py
+++ b/torchcomp/core.py
@@ -35,7 +35,6 @@ def compressor_cuda_kernel(
             at_mask[b, t] = 1
         else:
             coeff = rt_b
-            at_mask[b, t] = 0
         g *= 1 - coeff
         g += coeff * f
         y[b, t] = g