AstroBWT OpenCL code

src/backend/opencl/runners/OclAstroBWTRunner.h

@@ -0,0 +1,95 @@
+/* XMRig
+ * Copyright 2010      Jeff Garzik <jgarzik@pobox.com>
+ * Copyright 2012-2014 pooler      <pooler@litecoinpool.org>
+ * Copyright 2014      Lucas Jones <https://github.com/lucasjones>
+ * Copyright 2014-2016 Wolf9466    <https://github.com/OhGodAPet>
+ * Copyright 2016      Jay D Dee   <jayddee246@gmail.com>
+ * Copyright 2017-2018 XMR-Stak    <https://github.com/fireice-uk>, <https://github.com/psychocrypt>
+ * Copyright 2018-2020 SChernykh   <https://github.com/SChernykh>
+ * Copyright 2016-2020 XMRig       <https://github.com/xmrig>, <support@xmrig.com>
+ *
+ *   This program is free software: you can redistribute it and/or modify
+ *   it under the terms of the GNU General Public License as published by
+ *   the Free Software Foundation, either version 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   This program is distributed in the hope that it will be useful,
+ *   but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ *   GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License
+ *   along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifndef XMRIG_OCLASTROBWTRUNNER_H
+#define XMRIG_OCLASTROBWTRUNNER_H
+
+
+#include "backend/opencl/runners/OclBaseRunner.h"
+
+namespace xmrig {
+
+
+class AstroBWT_FilterKernel;
+class AstroBWT_MainKernel;
+class AstroBWT_PrepareBatch2Kernel;
+class AstroBWT_Salsa20Kernel;
+class AstroBWT_SHA3InitialKernel;
+class AstroBWT_SHA3Kernel;
+class AstroBWT_FindSharesKernel;
+
+
+class OclAstroBWTRunner : public OclBaseRunner
+{
+public:
+    static constexpr uint32_t BWT_DATA_MAX_SIZE = 600 * 1024 - 256;
+    static constexpr uint32_t BWT_DATA_STRIDE = (BWT_DATA_MAX_SIZE + 256 + 255) & ~255U;
+
+    XMRIG_DISABLE_COPY_MOVE_DEFAULT(OclAstroBWTRunner)
+
+    OclAstroBWTRunner(size_t index, const OclLaunchData &data);
+    ~OclAstroBWTRunner() override;
+
+    inline uint32_t roundSize() const override { return m_batch_size1; }
+
+    // ~0.5% of all hashes are invalid
+    inline uint32_t processedHashes() const override { return static_cast<uint32_t>(m_processedHashes * 0.995); }
+
+protected:
+    size_t bufferSize() const override;
+    void run(uint32_t nonce, uint32_t *hashOutput) override;
+    void set(const Job &job, uint8_t *blob) override;
+    void build() override;
+    void init() override;
+
+private:
+    AstroBWT_SHA3InitialKernel*   m_sha3_initial_kernel   = nullptr;
+    AstroBWT_SHA3Kernel*          m_sha3_kernel           = nullptr;
+    AstroBWT_Salsa20Kernel*       m_salsa20_kernel        = nullptr;
+    AstroBWT_MainKernel*          m_bwt_kernel            = nullptr;
+    AstroBWT_FilterKernel*        m_filter_kernel         = nullptr;
+    AstroBWT_PrepareBatch2Kernel* m_prepare_batch2_kernel = nullptr;
+    AstroBWT_FindSharesKernel*    m_find_shares_kernel    = nullptr;
+
+
+    cl_mem m_salsa20_keys                                 = nullptr;
+    cl_mem m_bwt_data                                     = nullptr;
+    cl_mem m_bwt_data_sizes                               = nullptr;
+    cl_mem m_indices                                      = nullptr;
+    cl_mem m_tmp_indices                                  = nullptr;
+    cl_mem m_filtered_hashes                              = nullptr;
+
+    uint32_t m_workgroup_size                             = 0;
+    uint64_t m_bwt_allocation_size                        = 0;
+    uint64_t m_batch_size1                                = 0;
+    uint32_t m_processedHashes                                = 0;
+
+    uint32_t* m_bwt_data_sizes_host                       = nullptr;
+};
+
+
+} /* namespace xmrig */
+
+
+#endif // XMRIG_OCLASTROBWTRUNNER_H