more set ups.

Author: JulioJerez

newton-4.00/sdk/dBrain/gpu/ndBrainGpuBuffer.cpp

@@ -20,7 +20,7 @@
 ndInt64 ndBrainGpuBuffer::m_memoryUsed = 0;
 
 ndBrainGpuBuffer::ndBrainGpuBuffer(ndBrainGpuContext* const context, ndInt64 sizeInByte, ndStorageBufferType bufferTypeFlags, ndDeviceBufferType deviceType)
-	:ndContainersFreeListAlloc<ndBrainGpuBuffer>()
+	:ndClassAlloc()
 	,m_context(context)
 	,m_sizeInBytes(size_t(sizeInByte))
 	,m_bufferType(bufferTypeFlags)

newton-4.00/sdk/dBrain/gpu/ndBrainGpuBuffer.h

@@ -32,29 +32,9 @@ enum ndDeviceBufferType
 	ndCpuMappable,
 };
 
-#if !defined (D_USE_VULKAN_SDK)
+#if defined (D_USE_VULKAN_SDK)
 
 class ndBrainGpuBuffer : public ndClassAlloc
-{
-	protected:
-	ndBrainGpuBuffer(ndBrainGpuContext* const, ndInt64, ndStorageBufferType, ndDeviceBufferType):m_sizeInBytes(0){}
-
-	public:
-	virtual ~ndBrainGpuBuffer() {}
-	void* GetBuffer() const { return nullptr;}
-	size_t SizeInBytes() const { return 0; }
-
-	virtual void LoadData(size_t sizeInBytes, const void* const inputData) = 0;
-	virtual void UnloadData(size_t sizeInBytes, void* const outputData) const = 0;
-
-	protected:
-	size_t  m_sizeInBytes;
-	friend class ndScopeMapBuffer;
-};
-
-#else
-
-class ndBrainGpuBuffer: public ndContainersFreeListAlloc<ndBrainGpuBuffer>
 {
 	public:
 	virtual ~ndBrainGpuBuffer();
@@ -79,6 +59,26 @@ class ndBrainGpuBuffer: public ndContainersFreeListAlloc<ndBrainGpuBuffer>
 	static ndInt64 m_memoryUsed;
 	friend class ndScopeMapBuffer;
 };
+
+#else
+
+class ndBrainGpuBuffer : public ndClassAlloc
+{
+	protected:
+	ndBrainGpuBuffer(ndBrainGpuContext* const, ndInt64, ndStorageBufferType, ndDeviceBufferType):m_sizeInBytes(0){}
+
+	public:
+	virtual ~ndBrainGpuBuffer() {}
+	void* GetBuffer() const { return nullptr;}
+	size_t SizeInBytes() const { return 0; }
+
+	virtual void LoadData(size_t sizeInBytes, const void* const inputData) = 0;
+	virtual void UnloadData(size_t sizeInBytes, void* const outputData) const = 0;
+
+	protected:
+	size_t  m_sizeInBytes;
+	friend class ndScopeMapBuffer;
+};
 #endif
 
 #endif

newton-4.00/sdk/dBrain/ndBrainAgentDeterministicPolicyGradient_Trainer.h

@@ -208,13 +208,12 @@ class ndBrainAgentDeterministicPolicyGradient_Trainer : public ndClassAlloc
 	ndBrainVector m_criticNextObservationActionBatch;
 
 	ndBrainVector m_nextQValue;
+	ndBrainVector m_expectedRewards;
 	ndBrainVector m_criticValue[ND_NUMBER_OF_CRITICS];
 	ndBrainVector m_rewardBatch[ND_NUMBER_OF_CRITICS];
-	//ndBrainVector m_criticTruthValue[ND_NUMBER_OF_CRITICS];
 	ndBrainVector m_criticInputGradients[ND_NUMBER_OF_CRITICS];
 	ndBrainVector m_criticOutputGradients[ND_NUMBER_OF_CRITICS];
 
-	ndBrainVector m_expectedRewards;
 	ndArray<ndInt32> m_miniBatchIndexBuffer;
 	ndBrainAgentDeterministicPolicyGradient_Agent::ndTrajectoryTransition m_replayBuffer;
 	ndBrainAgentDeterministicPolicyGradient_Agent* m_agent;