Update comments for workspace changes

Author: weiya711

include/taco/ir/workspace_rewriter.h

@@ -13,7 +13,9 @@ class Stmt;
 class Expr;
 }
 
-/// Simplifies a statement (e.g. by applying constant copy propagation).
+/// Rewrite a post-lowered IR statement to take into account multidimensional temporaries. 
+/// Replaces Dimension GetProperty nodes that correspond to temporary workspaces with 
+/// their corresponding dimension found in the temporarySizeMap. 
 ir::Stmt rewriteTemporaryGP(const ir::Stmt& stmt, std::vector<TensorVar> whereTemps,
                             std::map<TensorVar, std::vector<ir::Expr>> temporarySizeMap);
 

include/taco/lower/lowerer_impl_imperative.h

@@ -494,7 +494,10 @@ class LowererImplImperative : public LowererImpl {
   std::vector<TensorVar> whereTemps;
   std::map<TensorVar, const AccessNode *> whereTempsToResult;
 
+  // Map temporary tensorVars to a list of size expressions for each mode
   std::map<TensorVar, std::vector<ir::Expr>> temporarySizeMap;
+  
+  // List that contains all temporary tensorVars
   std::vector<TensorVar> temporaries;
 
   bool captureNextLocatePos = false;