Skip to content

Commit

Permalink
set buffer to visible
Browse files Browse the repository at this point in the history
  • Loading branch information
marty1885 committed Nov 25, 2019
1 parent 5e59d8d commit b30972d
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion Etaler/Backends/CPUBackend.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
namespace et
{

struct CPUBuffer : public BufferImpl
struct ETALER_EXPORT CPUBuffer : public BufferImpl
{
CPUBuffer(const Shape& shape, DType dtype, std::shared_ptr<Backend> backend)
: BufferImpl(shape.volume(), dtype, std::move(backend))
Expand Down
2 changes: 1 addition & 1 deletion Etaler/Backends/OpenCLBackend.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ struct ETALER_EXPORT OpenCLBuffer : public BufferImpl
cl::Buffer buffer_;
};

struct KernelManager
struct ETALER_EXPORT KernelManager
{
KernelManager() : KernelManager(cl::Device(), cl::Context()) {};
KernelManager(cl::Device device, cl::Context context);
Expand Down
4 changes: 2 additions & 2 deletions Etaler/Core/TensorImpl.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
namespace et
{

struct BufferImpl : public std::enable_shared_from_this<BufferImpl>
struct ETALER_EXPORT BufferImpl : public std::enable_shared_from_this<BufferImpl>
{
BufferImpl(size_t size, DType dtype, std::shared_ptr<Backend> backend)
: size_(size), dtype_(dtype), backend_(backend) {}
Expand All @@ -26,7 +26,7 @@ struct BufferImpl : public std::enable_shared_from_this<BufferImpl>
std::shared_ptr<Backend> backend_;
};

struct TensorImpl : public std::enable_shared_from_this<TensorImpl>
struct ETALER_EXPORT TensorImpl : public std::enable_shared_from_this<TensorImpl>
{
TensorImpl(std::shared_ptr<BufferImpl> buffer, Shape shape, Shape stride, size_t offset=0)
: buffer_(buffer), shape_(shape), stride_(stride), offset_(offset) {}
Expand Down

0 comments on commit b30972d

Please sign in to comment.