| .. |
|
test_all_gather.cc
|
impl distributed launch with NCCL (#106)
|
2023-09-05 09:47:35 +08:00 |
|
test_all_reduce.cc
|
impl distributed launch with NCCL (#106)
|
2023-09-05 09:47:35 +08:00 |
|
test_batch_norm.cc
|
refactor(core): 添加新的 `OpType` 定义 (#99)
|
2023-08-07 11:17:05 +08:00 |
|
test_broadcast.cc
|
impl distributed launch with NCCL (#106)
|
2023-09-05 09:47:35 +08:00 |
|
test_clip.cc
|
memory_allocator (#103)
|
2023-08-13 13:39:35 +08:00 |
|
test_concat.cc
|
support Dynamic tensor infer shape and fix memory pool (#176)
|
2023-11-23 13:11:50 +08:00 |
|
test_conv.cc
|
Add layer normalization (#181)
|
2023-11-24 15:15:14 +08:00 |
|
test_conv_transposed_2d.cc
|
ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61)
|
2023-03-27 21:28:49 +08:00 |
|
test_element_wise.cc
|
ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61)
|
2023-03-27 21:28:49 +08:00 |
|
test_expand.cc
|
框架支持bert/gpt2模型构图 (#94)
|
2023-08-29 16:06:52 +08:00 |
|
test_extend.cc
|
ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61)
|
2023-03-27 21:28:49 +08:00 |
|
test_gather.cc
|
框架支持bert/gpt2模型构图 (#94)
|
2023-08-29 16:06:52 +08:00 |
|
test_gather_elements.cc
|
Add GatherElements op and cuda kernel (#149)
|
2023-10-12 09:18:12 +08:00 |
|
test_matmul.cc
|
support mixed dtype (#102)
|
2023-08-16 21:49:43 +08:00 |
|
test_matmul_integer.cc
|
feat: add matmulinteger op
|
2023-12-19 14:14:33 +08:00 |
|
test_pad.cc
|
ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61)
|
2023-03-27 21:28:49 +08:00 |
|
test_pooling.cc
|
Pooling ceil mode (#155)
|
2023-10-09 20:51:39 +08:00 |
|
test_reduce.cc
|
Add ReduceSum op and kernel (#160)
|
2023-11-24 09:29:58 +08:00 |
|
test_reshape.cc
|
Cpu backend2 (#77)
|
2023-04-17 12:15:23 +08:00 |
|
test_resize.cc
|
ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61)
|
2023-03-27 21:28:49 +08:00 |
|
test_sendrecv.cc
|
Add send and recv operators based on NCCL (#182)
|
2023-12-14 16:38:03 +08:00 |
|
test_slice.cc
|
Dev for 202303ddl (#66)
|
2023-04-18 15:10:33 +08:00 |
|
test_split.cc
|
impl sqrt on CUDA (#109)
|
2023-08-18 12:17:47 +08:00 |
|
test_transpose.cc
|
support mixed dtype (#102)
|
2023-08-16 21:49:43 +08:00 |
|
test_unary.cc
|
add cuda cast & support half-precision for gather
|
2023-12-14 11:24:25 +08:00 |
|
test_where.cc
|
add where fp16
|
2023-12-08 16:57:49 +08:00 |
97e3377ca5