diff --git a/commonLibs/attachmentSupport/CMakeLists.txt b/commonLibs/attachmentSupport/CMakeLists.txt
index bdf8b01..1ce5dc2 100644
--- a/commonLibs/attachmentSupport/CMakeLists.txt
+++ b/commonLibs/attachmentSupport/CMakeLists.txt
@@ -1,4 +1,5 @@
 add_library(attachmentSupport SHARED
+	compute.cpp
 	stimulusProducer.cpp
 	stagingBuffer.cpp
 )
diff --git a/commonLibs/attachmentSupport/compute.cpp b/commonLibs/attachmentSupport/compute.cpp
new file mode 100644
index 0000000..3e848b7
--- /dev/null
+++ b/commonLibs/attachmentSupport/compute.cpp
@@ -0,0 +1,147 @@
+#include <user/compute.h>
+#include <stdexcept>
+#include <string>
+#include <vector>
+#include <iostream>
+#include <string_view>
+
+namespace smo {
+namespace compute {
+
+// Helper function to parse OpenCL version string
+static std::pair<int, int> parseOpenClVersion(const std::string& versionStr)
+{
+	size_t spacePos = versionStr.find(' ');
+	if (spacePos == std::string::npos) { return {-1, -1}; }
+
+	std::string versionNum = versionStr.substr(spacePos + 1);
+	size_t dotPos = versionNum.find('.');
+	if (dotPos == std::string::npos) { return {-1, -1}; }
+
+	try {
+		int major = std::stoi(versionNum.substr(0, dotPos));
+		int minor = std::stoi(versionNum.substr(dotPos + 1));
+		return {major, minor};
+	} catch (const std::exception&) {
+		return {-1, -1};
+	}
+}
+
+// Implementation of validateOpenClVersion (declared in user/compute.h)
+bool validateOpenClVersion(
+	std::string_view versionStr, std::string_view versionType,
+	int minMajor, int minMinor)
+{
+	auto [major, minor] = parseOpenClVersion(std::string(versionStr));
+
+	if (major == -1 && minor == -1)
+	{
+		std::cerr << __func__ << ": failed to parse OpenCL " << versionType
+			<< " version: " << versionStr << std::endl;
+		return false;
+	}
+
+	if (major < minMajor || (major == minMajor && minor < minMinor))
+	{
+		std::cerr << __func__ << ": OpenCL " << versionType << " version "
+			<< major << "." << minor << " found, but " << minMajor << "."
+			<< minMinor << " or higher is required" << std::endl;
+		return false;
+	}
+
+	std::cout << __func__ << ": OpenCL " << versionType << " version: "
+		<< versionStr << std::endl;
+	return true;
+}
+
+ComputeDevice::ComputeDevice(cl_platform_id platformId, cl_device_id deviceId)
+: platform(platformId), device(deviceId),
+context(nullptr), commandQueue(nullptr)
+{
+	cl_int err;
+
+	// Create context for this device
+	context = clCreateContext(
+		nullptr, 1, &device,
+		nullptr, nullptr, &err);
+
+	if (err != CL_SUCCESS || !context)
+	{
+		throw std::runtime_error(
+			std::string(__func__) + ": failed to create context for device: " +
+			std::to_string(err));
+	}
+
+	// Create command queue
+	cl_command_queue_properties queueProps = 0;
+	commandQueue = clCreateCommandQueue(
+		context, device, queueProps, &err);
+
+	if (err != CL_SUCCESS || !commandQueue)
+	{
+		clReleaseContext(context);
+		context = nullptr;
+		throw std::runtime_error(
+			std::string(__func__) + ": failed to create command queue for "
+			"device: " + std::to_string(err));
+	}
+}
+
+ClBuffer::ClBuffer(void* hostPtr, size_t size, cl_mem_flags flags,
+	const std::vector<std::shared_ptr<ComputeDevice>>& devices)
+	: hostPtr(hostPtr), size(size), flags(flags)
+{
+	associations.reserve(devices.size());
+
+	// Create a buffer for each device's context
+	for (const auto& device : devices)
+	{
+		if (!device->context) { continue; }
+
+		cl_int err;
+		cl_mem_flags bufferFlags = CL_MEM_USE_HOST_PTR | flags;
+		cl_mem buffer = clCreateBuffer(
+			device->context,
+			bufferFlags,
+			size, hostPtr,
+			&err);
+
+		if (err != CL_SUCCESS || !buffer)
+		{
+			// Release any buffers already created before throwing
+			for (auto& assoc : associations)
+			{
+				if (assoc.buffer) {
+					clReleaseMemObject(assoc.buffer);
+				}
+			}
+			throw std::runtime_error(
+				std::string(__func__) + ": failed to create buffer for "
+				"device: " + std::to_string(err));
+		}
+
+		associations.emplace_back(buffer, device);
+	}
+}
+
+cl_mem ClBuffer::getAssociatedBufferHandleForDevice(
+	const std::shared_ptr<ComputeDevice>& device) const
+{
+	if (!device)
+	{
+		throw std::invalid_argument(std::string(__func__)
+			+ ": device is nullptr");
+	}
+
+	for (const auto& assoc : associations)
+	{
+		if (assoc.device == device) {
+			return assoc.buffer;
+		}
+	}
+
+	return nullptr;
+}
+
+} // namespace compute
+} // namespace smo
diff --git a/include/user/compute.h b/include/user/compute.h
index f956b67..21a8035 100644
--- a/include/user/compute.h
+++ b/include/user/compute.h
@@ -3,12 +3,18 @@
 
 #include <memory>
 #include <vector>
+#include <string_view>
 #define CL_TARGET_OPENCL_VERSION 120
 #include <CL/cl.h>
 
 namespace smo {
 namespace compute {
 
+// Helper function to validate OpenCL version
+bool validateOpenClVersion(
+	std::string_view versionStr, std::string_view versionType,
+	int minMajor, int minMinor);
+
 /**
  * @brief OpenCL compute device information
  *
@@ -108,6 +114,14 @@ public:
 	ClBuffer(const ClBuffer&) = delete;
 	ClBuffer& operator=(const ClBuffer&) = delete;
 
+	/**
+	 * @brief Get the cl_mem handle for a specific compute device
+	 * @param device The compute device to find the buffer for
+	 * @return The cl_mem handle for the device, or nullptr if not found
+	 */
+	cl_mem getAssociatedBufferHandleForDevice(
+		const std::shared_ptr<ComputeDevice>& device) const;
+
 	void* hostPtr;
 	size_t size;
 	cl_mem_flags flags;
diff --git a/smocore/computeManager/computeManager.cpp b/smocore/computeManager/computeManager.cpp
index 46870b6..83e3a0e 100644
--- a/smocore/computeManager/computeManager.cpp
+++ b/smocore/computeManager/computeManager.cpp
@@ -1,92 +1,14 @@
-#include <computeManager/computeManager.h>
 #include <iostream>
 #include <stdexcept>
 #include <string>
 #include <string_view>
 #include <algorithm>
+#include <computeManager/computeManager.h>
+#include <user/compute.h>
 
 namespace smo {
 namespace compute {
 
-// Helper function to parse OpenCL version string
-static std::pair<int, int> parseOpenClVersion(const std::string& versionStr)
-{
-	size_t spacePos = versionStr.find(' ');
-	if (spacePos == std::string::npos) { return {-1, -1}; }
-
-	std::string versionNum = versionStr.substr(spacePos + 1);
-	size_t dotPos = versionNum.find('.');
-	if (dotPos == std::string::npos) { return {-1, -1}; }
-
-	try {
-		int major = std::stoi(versionNum.substr(0, dotPos));
-		int minor = std::stoi(versionNum.substr(dotPos + 1));
-		return {major, minor};
-	} catch (const std::exception&) {
-		return {-1, -1};
-	}
-}
-
-// Helper function to validate OpenCL version
-static bool validateOpenClVersion(
-	std::string_view versionStr, std::string_view versionType,
-	int minMajor, int minMinor)
-{
-	auto [major, minor] = parseOpenClVersion(std::string(versionStr));
-
-	if (major == -1 && minor == -1)
-	{
-		std::cerr << __func__ << ": failed to parse OpenCL " << versionType
-			<< " version: " << versionStr << std::endl;
-		return false;
-	}
-
-	if (major < minMajor || (major == minMajor && minor < minMinor))
-	{
-		std::cerr << __func__ << ": OpenCL " << versionType << " version "
-			<< major << "." << minor << " found, but " << minMajor << "."
-			<< minMinor << " or higher is required" << std::endl;
-		return false;
-	}
-
-	std::cout << __func__ << ": OpenCL " << versionType << " version: "
-		<< versionStr << std::endl;
-	return true;
-}
-
-ComputeDevice::ComputeDevice(cl_platform_id platformId, cl_device_id deviceId)
-: platform(platformId), device(deviceId),
-context(nullptr), commandQueue(nullptr)
-{
-	cl_int err;
-
-	// Create context for this device
-	context = clCreateContext(
-		nullptr, 1, &device,
-		nullptr, nullptr, &err);
-
-	if (err != CL_SUCCESS || !context)
-	{
-		throw std::runtime_error(
-			std::string(__func__) + ": failed to create context for device: " +
-			std::to_string(err));
-	}
-
-	// Create command queue
-	cl_command_queue_properties queueProps = 0;
-	commandQueue = clCreateCommandQueue(
-		context, device, queueProps, &err);
-
-	if (err != CL_SUCCESS || !commandQueue)
-	{
-		clReleaseContext(context);
-		context = nullptr;
-		throw std::runtime_error(
-			std::string(__func__) + ": failed to create command queue for "
-			"device: " + std::to_string(err));
-	}
-}
-
 void ComputeManager::initialize()
 {
 	if (initialized) { return; }
@@ -226,43 +148,6 @@ void ComputeManager::finalize()
 	std::cout << __func__ << ": Finalized" << std::endl;
 }
 
-ClBuffer::ClBuffer(void* hostPtr, size_t size, cl_mem_flags flags,
-	const std::vector<std::shared_ptr<ComputeDevice>>& devices)
-	: hostPtr(hostPtr), size(size), flags(flags)
-{
-	associations.reserve(devices.size());
-
-	// Create a buffer for each device's context
-	for (const auto& device : devices)
-	{
-		if (!device->context) { continue; }
-
-		cl_int err;
-		cl_mem_flags bufferFlags = CL_MEM_USE_HOST_PTR | flags;
-		cl_mem buffer = clCreateBuffer(
-			device->context,
-			bufferFlags,
-			size, hostPtr,
-			&err);
-
-		if (err != CL_SUCCESS || !buffer)
-		{
-			// Release any buffers already created before throwing
-			for (auto& assoc : associations)
-			{
-				if (assoc.buffer) {
-					clReleaseMemObject(assoc.buffer);
-				}
-			}
-			throw std::runtime_error(
-				std::string(__func__) + ": failed to create buffer for "
-				"device: " + std::to_string(err));
-		}
-
-		associations.emplace_back(buffer, device);
-	}
-}
-
 std::shared_ptr<ClBuffer>
 ComputeManager::createUseHostPtrBuffer(
 	void* hostPtr, size_t size, cl_mem_flags flags)