123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141 |
- #ifndef EASYBANG_RESIZE_H_
- #define EASYBANG_RESIZE_H_
- #include <string>
- #include "cnrt.h"
- #include "cxxutil/edk_attribute.h"
- #include "cxxutil/exception.h"
- namespace edk {
- class MluResizePrivate;
- /**
- * @brief Mlu resize operator helper class
- */
- class MluResize {
- public:
-
- MluResize();
-
- ~MluResize();
-
- struct Attr {
-
- uint32_t src_w{0}, src_h{0};
-
- uint32_t src_stride_y{0}, src_stride_uv{0};
-
- uint32_t dst_w{0}, dst_h{0};
-
- int batch_size = 1;
- int core = 4;
- uint32_t channel_id = 0;
- };
-
- void SetMluQueue(cnrtQueue_t queue, bool exclusive = false);
-
- void DestroyMluQueue();
-
- cnrtQueue_t GetMluQueue() const;
-
- bool Init(const Attr& attr);
-
- const Attr& GetAttr();
-
- void Destroy();
-
- std::string GetLastError() const;
-
- void BatchingUp(void* src_y, void* src_uv);
-
- bool SyncOneOutput(void* dst_y, void* dst_uv);
- private:
- MluResizePrivate* d_ptr_ = nullptr;
- MluResize(const MluResize&) = delete;
- MluResize& operator=(const MluResize&) = delete;
- };
- }
- #endif
|