-
Notifications
You must be signed in to change notification settings - Fork 104
/
kernel_depends.toml
executable file
·48 lines (42 loc) · 1.69 KB
/
kernel_depends.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
[mlu-ops]
[kernels]
# # mlu-ops API dependencies: only kernels/<> compile dependencies
# ## format:
# ## <kernel_name> = ["op_1", "op_2"]
abs = ["tensor_stride_process","unary_op"]
div = ["binary_op","tensor_stride_process"]
log = ["unary_op","tensor_stride_process"]
lgamma = ["unary_op","tensor_stride_process"]
sqrt = ["binary_op", "unary_op", "tensor_stride_process"]
carafe = ["tensor_stride_process"]
[gtest]
# # mlu_op_gtest API dependencies under mlu_op_gtest/pb_gtest/src/zoo/<op>
# ## If mlu_op_gtest use dlopen to call mluops, dependencies below may not be necessary
fft =["tensor_stride_process"]
psroipool_backward = ["psroipool"]
psroipool_forward = ["psroipool"]
roi_align_rotated_forward = ["roi_align_rotated"]
roi_align_rotated_backward = ["roi_align_rotated"]
roiaware_pool3d_forward = ["roiaware_pool3d"]
roiaware_pool3d_backward = ["roiaware_pool3d"]
roi_crop_backward = ["roi_crop"]
roi_crop_forward = ["roi_crop"]
rotated_feature_align_backward = ["rotated_feature_align"]
rotated_feature_align_forward = ["rotated_feature_align"]
sqrt_backward = ["sqrt"]
tin_shift_backward = ["tin_shift"]
tin_shift_forward = ["tin_shift"]
psamask_forward = ["psamask"]
psamask_backward = ["psamask"]
focal_loss_sigmoid_forward = ["focal_loss_sigmoid"]
focal_loss_sigmoid_backward = ["focal_loss_sigmoid"]
three_interpolate_forward = ["three_interpolate"]
three_interpolate_backward = ["three_interpolate"]
active_rotated_filter_forward = ["active_rotated_filter"]
deform_roi_pool_forward = ["deform_roi_pool"]
deform_roi_pool_backward = ["deform_roi_pool"]
carafe_forward = ["carafe"]
carafe_backward = ["carafe"]
dcn_backward_weight = ["dcn_common"]
dcn_backward_data = ["dcn_common"]
dcn_forward = ["dcn_common"]