atb/operation_infra.h
Definition
class OperationInfra : public Operation {
public:
OperationInfra();
OperationInfra(const OperationInfra &other);
OperationInfra& operator = (const OperationInfra &other);
~OperationInfra();
void SetExecuteStreamId(uint32_t streamId);
uint32_t GetExecuteStreamId() const;
aclrtStream GetExecuteStream(Context *context);
private:
std::unique_ptr<OperationImpl> impl_;
};
Parameters
Member |
Parameter |
Return Value |
Description |
|---|---|---|---|
OperationInfra |
- |
- |
Constructor. |
OperationInfra |
other |
- |
Copy constructor.
|
operator = |
other |
OperationInfra& |
Assignment constructor.
Successful execution returns the object itself. |
~OperationInfra |
- |
- |
Destructor. |
SetExecuteStreamId |
streamId |
- |
Used to set the ID of the stream used by the operation. The ID is the stream sequence number in the context. This API must be used together with the GetExecuteStream API.
|
GetExecuteStreamId |
- |
uint32_t |
Used to obtain the ID of the stream used by the current operation. Successful execution returns the ID of the stream. The ID is the stream sequence number in the context. |
GetExecuteStream |
context |
aclrtStream |
Used to obtain the stream used by the current operation.
If the execution is successful, the stream used by the current operation is returned. |