|
std::unordered_map< numerics::TensorHashType, std::shared_ptr< talsh::Tensor > > | tensors_ |
|
std::unordered_map< TensorOpExecHandle, std::shared_ptr< talsh::TensorTask > > | tasks_ |
|
◆ execute()
Executes the tensor operation found in a DAG node asynchronously, returning the execution handle in exec_handle that can later be used for testing for completion of the operation execution. Returns an integer error code (0:Success).
Implements exatn::runtime::TensorNodeExecutor.
◆ getLocalTensor()
std::shared_ptr< talsh::Tensor > exatn::runtime::TalshNodeExecutor::getLocalTensor |
( |
const numerics::Tensor & |
tensor, |
|
|
const std::vector< std::pair< DimOffset, DimExtent >> & |
slice_spec |
|
) |
| |
|
overridevirtual |
◆ initialize()
void exatn::runtime::TalshNodeExecutor::initialize |
( |
| ) |
|
|
overridevirtual |
◆ sync()
bool exatn::runtime::TalshNodeExecutor::sync |
( |
TensorOpExecHandle |
op_handle, |
|
|
int * |
error_code, |
|
|
bool |
wait = false |
|
) |
| |
|
overridevirtual |
◆ talsh_initialized_
bool exatn::runtime::TalshNodeExecutor::talsh_initialized_ = false |
|
staticprotected |
TAL-SH initialization status
◆ talsh_node_exec_count_
int exatn::runtime::TalshNodeExecutor::talsh_node_exec_count_ = 0 |
|
staticprotected |
Number of instances of TAL-SH node executors
◆ tasks_
std::unordered_map<TensorOpExecHandle,std::shared_ptr<talsh::TensorTask> > exatn::runtime::TalshNodeExecutor::tasks_ |
|
protected |
Active execution handles associated with tensor operations currently executed by TAL-SH
◆ tensors_
std::unordered_map<numerics::TensorHashType,std::shared_ptr<talsh::Tensor> > exatn::runtime::TalshNodeExecutor::tensors_ |
|
protected |
The documentation for this class was generated from the following files:
- /Users/aqw/exatn/src/runtime/executor/node_executors/talsh/node_executor_talsh.hpp
- /Users/aqw/exatn/src/runtime/executor/node_executors/talsh/node_executor_talsh.cpp