Commit Graph

9 Commits (bc7a3afa687696541b032d56d1e9a8ca8e101c77)

Author SHA1 Message Date
Leo Chen a93488839d
fix compile problem (#31850)
4 years ago
Leo Chen d1a4c53eee
[NPU] support default stream (#31510)
4 years ago
Leo Chen b3446670c1
[NPU] add npu kernel for concat op (#31695)
4 years ago
Leo Chen 03803f20fd
[NPU] support list of tensor input (#31801)
4 years ago
Leo Chen 0310945f5c
[NPU] Support npu op layer_norm and layer_norm_grad (#31310)
4 years ago
Leo Chen d23bf89cf6
support list of list attribute for NPU (#31299)
4 years ago
Leo Chen 5cb20f30fc
add npu kernel for elementwise_sub and elementwise_sub_grad (#30973)
4 years ago
Leo Chen 1201cd2ef2
[feature] support npu allocator, part 2 (#30972)
4 years ago
Leo Chen 7e049108c5
[feature] support npu operator (#30951)
4 years ago