|
|
|
@ -50,7 +50,7 @@ class AiCoreOpTask {
|
|
|
|
|
|
|
|
|
|
uint32_t GetBlockDim() const {return block_dim_;}
|
|
|
|
|
|
|
|
|
|
void SetSingleOp(bool is_single_op) {is_single_op_ = is_single_op};
|
|
|
|
|
void SetSingleOp(bool is_single_op) {is_single_op_ = is_single_op;};
|
|
|
|
|
|
|
|
|
|
protected:
|
|
|
|
|
Status UpdateTilingInfo(TaskContext &context);
|
|
|
|
@ -74,7 +74,7 @@ class AiCoreOpTask {
|
|
|
|
|
uint32_t args_size_ = 0;
|
|
|
|
|
uint32_t block_dim_ = 1;
|
|
|
|
|
bool clear_atomic_ = true;
|
|
|
|
|
bool is_single_op_ false;
|
|
|
|
|
bool is_single_op_ = false;
|
|
|
|
|
std::vector<int> output_indices_to_skip_;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|