Package: Major fall update:
-> Major performance improvement of CONV for small batches -> Adopted NCHW image layout -> 3-D convolutions supported -> HFMA2 deprecated -> Multi-threaded auto-tuning -> Tensorflow Wrapper
This commit is contained in:
@@ -38,14 +38,14 @@ class Buffer: public HandleInterface<Buffer, CUdeviceptr>
|
||||
{
|
||||
public:
|
||||
Buffer(Context const & context, size_t size);
|
||||
Buffer(Context const & context, CUdeviceptr cu, bool take_ownership);
|
||||
void set_zero(Stream const & queue, size_t size);
|
||||
void set_zero(Stream const & queue);
|
||||
Handle<CUdeviceptr> const & cu() const;
|
||||
Handle<CUdeviceptr> & cu();
|
||||
|
||||
private:
|
||||
Context context_;
|
||||
Handle<CUdeviceptr> cu_;
|
||||
size_t size_;
|
||||
};
|
||||
|
||||
}
|
||||
|
Reference in New Issue
Block a user