refactor

DiffAPF · Mar 25, 2024 · ef54ae9 · ef54ae9
1 parent 8018c47
commit ef54ae9
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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