adstraw commented on code in PR #12411: URL: https://github.com/apache/tvm/pull/12411#discussion_r949709289
########## src/runtime/hexagon/hexagon_user_dma.h: ########## @@ -0,0 +1,90 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ + +#include <vector> + +#ifndef TVM_RUNTIME_HEXAGON_HEXAGON_USER_DMA_H_ +#define TVM_RUNTIME_HEXAGON_HEXAGON_USER_DMA_H_ + +namespace tvm { +namespace runtime { +namespace hexagon { + +#define DMA_SUCCESS 0 +#define DMA_FAILURE -1 + +class HexagonUserDMA { + public: + /*! + * \brief Initiate DMA to copy memory from source to destination address + * \param dst Destination address + * \param src Source address + * \param length Length in bytes to copy + * \returns Status, either DMA_SUCCESS or DMA_FAILURE + */ + int Copy(void* dst, void* src, uint32_t length); + + /*! + * \brief Wait until the number of DMAs in flight is less than or equal to some maximum + * \param max_dmas_in_flight Maximum number of DMAs allowed to be in flight + * to satisfy the `Wait` e.g. use `Wait(0)` to wait on "all" outstanding DMAs to complete + */ + void Wait(uint32_t max_dmas_in_flight); + + /*! + * \brief Poll the number of DMAs in flight + * \returns Number of DMAs in flight + */ + uint32_t Poll(); + + //! HexagonUserDMA uses the singleton pattern + static HexagonUserDMA& Get() { + static HexagonUserDMA* hud = new HexagonUserDMA(); + return *hud; + } + + private: + HexagonUserDMA() = default; + ~HexagonUserDMA(); + HexagonUserDMA(const HexagonUserDMA&) = delete; + HexagonUserDMA& operator=(const HexagonUserDMA&) = delete; + HexagonUserDMA(HexagonUserDMA&&) = delete; + HexagonUserDMA& operator=(HexagonUserDMA&&) = delete; + + //! \brief Initializes / resets the Hexagon User DMA engine + unsigned int Init(); + + //! \brief Calculates the number of DMAs in flight + uint32_t DMAsInFlight(); + + //! \brief Stores descriptors for all DMAs + std::vector<void*> dma_descriptors_; Review Comment: Possible issue - @csullivan informs me that the destructor may not get called for Hexagon runtime classes given that we kill the RPC to avoid zombie processes. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: commits-unsubscr...@tvm.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org
