|
16 | 16 | #include "RulesDNN/DNNAPIMigration.h" |
17 | 17 | #include "RuleInfra/ExprAnalysis.h" |
18 | 18 | #include "RulesMathLib/FFTAPIMigration.h" |
| 19 | +#include "RulesMathLib/RandomAPIMigration.h" |
| 20 | +#include "RulesMathLib/SolverAPIMigration.h" |
19 | 21 | #include "CodePin/GenCodePinHeader.h" |
20 | 22 | #include "RulesSecurity/Homoglyph.h" |
21 | 23 | #include "RulesLangLib/LIBCUAPIMigration.h" |
@@ -155,31 +157,29 @@ REGISTER_RULE(AssertRule, PassKind::PK_Migration) |
155 | 157 | REGISTER_RULE(GraphRule, PassKind::PK_Migration) |
156 | 158 | REGISTER_RULE(GraphicsInteropRule, PassKind::PK_Migration) |
157 | 159 |
|
158 | | -// |
159 | 160 | REGISTER_RULE(BLASEnumsRule, PassKind::PK_Migration, RuleGroupKind::RK_BLas) |
160 | | -REGISTER_RULE(RandomEnumsRule, PassKind::PK_Migration, RuleGroupKind::RK_Rng) |
| 161 | +REGISTER_RULE(BLASFunctionCallRule, PassKind::PK_Migration,RuleGroupKind::RK_BLas) |
| 162 | + |
161 | 163 | REGISTER_RULE(SPBLASEnumsRule, PassKind::PK_Migration, RuleGroupKind::RK_Sparse) |
162 | 164 | REGISTER_RULE(SPBLASFunctionCallRule, PassKind::PK_Migration,RuleGroupKind::RK_Sparse) |
| 165 | + |
| 166 | +REGISTER_RULE(RandomEnumsRule, PassKind::PK_Migration, RuleGroupKind::RK_Rng) |
163 | 167 | REGISTER_RULE(RandomFunctionCallRule, PassKind::PK_Migration,RuleGroupKind::RK_Rng) |
164 | 168 | REGISTER_RULE(DeviceRandomFunctionCallRule, PassKind::PK_Migration,RuleGroupKind::RK_Rng) |
165 | | -REGISTER_RULE(BLASFunctionCallRule, PassKind::PK_Migration,RuleGroupKind::RK_BLas) |
| 169 | + |
166 | 170 | REGISTER_RULE(SOLVEREnumsRule, PassKind::PK_Migration, RuleGroupKind::RK_Solver) |
167 | 171 | REGISTER_RULE(SOLVERFunctionCallRule, PassKind::PK_Migration,RuleGroupKind::RK_Solver) |
168 | 172 |
|
169 | | -// |
170 | 173 | REGISTER_RULE(LIBCURule, PassKind::PK_Migration, RuleGroupKind::RK_Libcu) |
171 | | -// |
| 174 | + |
172 | 175 | REGISTER_RULE(ThrustAPIRule, PassKind::PK_Migration, RuleGroupKind::RK_Thrust) |
173 | 176 | REGISTER_RULE(ThrustTypeRule, PassKind::PK_Migration, RuleGroupKind::RK_Thrust) |
174 | | -// Rules NCCL |
| 177 | + |
175 | 178 | REGISTER_RULE(ManualMigrateEnumsRule, PassKind::PK_Migration, RuleGroupKind::RK_NCCL) |
176 | 179 | REGISTER_RULE(NCCLRule, PassKind::PK_Migration, RuleGroupKind::RK_NCCL) |
177 | 180 |
|
178 | | - |
179 | | -// Rules FFT |
180 | 181 | REGISTER_RULE(FFTEnumsRule, PassKind::PK_Migration, RuleGroupKind::RK_FFT) |
181 | 182 | REGISTER_RULE(FFTFunctionCallRule, PassKind::PK_Migration,RuleGroupKind::RK_FFT) |
182 | 183 |
|
183 | | -// Rules DNN |
184 | 184 | REGISTER_RULE(CuDNNTypeRule, PassKind::PK_Migration, RuleGroupKind::RK_DNN) |
185 | 185 | REGISTER_RULE(CuDNNAPIRule, PassKind::PK_Migration, RuleGroupKind::RK_DNN) |
0 commit comments