Commit effa5a2f authored by Dmitry I. Lyakh's avatar Dmitry I. Lyakh

Finished implementation of reduced shaping in TAL-SH node executor.

parent 79dc9123
/** ExaTN::Numerics: Tensor operation
REVISION: 2020/06/25
REVISION: 2020/06/26
Copyright (C) 2018-2020 Dmitry I. Lyakh (Liakh)
Copyright (C) 2018-2020 Oak Ridge National Laboratory (UT-Battelle) **/
......@@ -214,7 +214,7 @@ std::string TensorOperation::getIndexPatternReduced() const
if(tensor.getDimExtent(i++) > 1){
++iter;
}else{
iter = indices.erase(iter);
iter = indices.erase(iter); //remove indices associated with extent-1 dimensions
}
}
tensors[symb_pos_[oprnd]] = assemble_symbolic_tensor(tensor_name,indices,conj);
......
......@@ -127,8 +127,8 @@ protected:
in an active tensor operation, tensor prefetch or tensor eviction. **/
bool tensorIsCurrentlyInUse(const talsh::Tensor * talsh_tens) const;
/** Maps generic exatn::numerics::Tensor to its TAL-SH implementation talsh::Tensor **/
std::unordered_map<numerics::TensorHashType,std::shared_ptr<talsh::Tensor>> tensors_;
/** Maps generic exatn::numerics::Tensor to its TAL-SH implementation **/
std::unordered_map<numerics::TensorHashType,TensorImpl> tensors_;
/** Active execution handles associated with tensor operations currently executed by TAL-SH **/
std::unordered_map<TensorOpExecHandle,std::shared_ptr<talsh::TensorTask>> tasks_;
/** Active tensor operand prefetching to accelerators tasks **/
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment