Expose RANDOMX_SCRATCHPAD_L3 as a function

NIFs that link with librandomx can't rely on configuration.h
ArweaveTeam · Dec 3, 2024 · eef4dc8 · eef4dc8
1 parent 0e5b2ac
commit eef4dc8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/randomx.cpp b/src/randomx.cpp
@@ -183,6 +183,10 @@ extern "C" {
 		return DatasetItemCount;
 	}
 
+	unsigned long randomx_get_scratchpad_size() {
+		return randomx::ScratchpadSize;
+	}
+
 	void randomx_init_dataset(randomx_dataset *dataset, randomx_cache *cache, unsigned long startItem, unsigned long itemCount) {
 		assert(dataset != nullptr);
 		assert(cache != nullptr);

diff --git a/src/randomx.h b/src/randomx.h
@@ -174,6 +174,13 @@ RANDOMX_EXPORT void *randomx_get_dataset_memory(randomx_dataset *dataset);
 */
 RANDOMX_EXPORT void randomx_release_dataset(randomx_dataset *dataset);
 
+/**
+ * Gets size of the scratchpad in bytes.
+ *
+ * @return the size of the scratchpad in bytes.
+*/
+RANDOMX_EXPORT unsigned long randomx_get_scratchpad_size(void);
+
 /**
  * Creates and initializes a RandomX virtual machine.
  *