Skip to content

support generic device

Latest
Latest

Choose a tag to compare

View all tags
@lgarithm lgarithm released this 06 Feb 22:18
· 5 commits to master since this release
v0.2.0
7707013
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode. 
make layers support cuda tensor (#78)

* invoke (#77)

* ASSERT_EQ

* using D = typename T0::device_type;

* fix test

* add <typename D> to layers

* simplify example

* use tensor_view as layer input

* simplify

* fix

* fix build

* concat kernel

* fix build

* io kernel

* fix include

* batch_norm kernel

* pool_traits

* pool kernel

* fix include

* typename D

* normalizer

* make it nvcc compatiable

* make it nvcc compatiable

* fix include

* make it nvcc compatiable

* stdtensor -> v0.6.1
Assets 2
Loading