test propagation using simple buffer casting transform test

Author: Ali Alshaarawy

thunder/tests/test_transforms.py

@@ -653,10 +653,9 @@ def test_buffer_quantization():
 
     from typing import Any, Optional, Tuple, Union, List
 
-    class QuantizeBuffers(thunder.core.transform_common.Transform):
+    class CastBuffers(thunder.core.transform_common.Transform):
         def __init__(self):
             self.quant_states = {}
-            self.quantized_submodule_names = set()
 
         def transform_module(self, model: thunder.ThunderModule):
             self.thunder_module = model
@@ -811,7 +810,7 @@ def forward(self, k: torch.Tensor, v: torch.Tensor) -> Tuple[torch.Tensor, torch
 
     k = torch.randn(2, 3, 4, 5, device=device, dtype=torch.half)
     v = torch.randn(2, 3, 4, 5, device=device, dtype=torch.half)
-    cast_jit = thunder.jit(model, transforms=[QuantizeBuffers(),])
+    cast_jit = thunder.jit(model, transforms=[CastBuffers(),])
     output_k, output_v = cast_jit(k, v)
 
     def check_dtypes(bsym):