Clear cached FSA properties when it labels are changed.

k2-fsa · Nov 8, 2021 · de5b200 · de5b200
1 parent ceb8e50
commit de5b200
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/k2/torch/csrc/fsa_class.cu b/k2/torch/csrc/fsa_class.cu
@@ -131,6 +131,7 @@ void FsaClass::SetLabels(torch::Tensor labels) {
   K2_CHECK_EQ(labels.scalar_type(), torch::kInt32);
   K2_CHECK(ContextFromTensor(labels)->IsCompatible(*fsa.Context()));
   Labels().copy_(labels);
+  properties = 0;  // Clear cached properties as we changed the labels
 }
 
 void FsaClass::SetAttr(const std::string &name, torch::IValue value) {