|
40 | 40 | */
|
41 | 41 | package com.oracle.graal.python.processor;
|
42 | 42 |
|
| 43 | +import static com.oracle.graal.python.processor.ConverterFactory.CLINIC_PACKAGE; |
| 44 | + |
43 | 45 | import java.io.IOException;
|
44 | 46 | import java.time.LocalDateTime;
|
45 | 47 | import java.util.ArrayList;
|
|
77 | 79 | import com.oracle.graal.python.processor.ArgumentClinicModel.BuiltinClinicData;
|
78 | 80 | import com.oracle.graal.python.processor.CodeWriter.Block;
|
79 | 81 |
|
80 |
| -import static com.oracle.graal.python.processor.ConverterFactory.CLINIC_PACKAGE; |
81 |
| - |
82 | 82 | public class ArgumentClinicProcessor extends AbstractProcessor {
|
83 | 83 | private static final boolean LOGGING = false;
|
84 | 84 | private static final String BuiltinAnnotationClass = "com.oracle.graal.python.builtins.Builtin";
|
@@ -148,12 +148,14 @@ private void writeCode(HashMap<TypeElement, Set<BuiltinClinicData>> enclosingTyp
|
148 | 148 | private void writeImports(CodeWriter w, Entry<TypeElement, Set<BuiltinClinicData>> enclosingType) throws IOException {
|
149 | 149 | log("Writing imports...");
|
150 | 150 | TreeSet<String> imports = new TreeSet<>();
|
151 |
| - imports.add("com.oracle.graal.python.nodes.function.PythonBuiltinBaseNode"); |
152 | 151 | imports.add(CLINIC_PACKAGE + ".ArgumentClinicProvider");
|
153 |
| - imports.add(CLINIC_PACKAGE + ".ArgumentCastNode"); |
154 | 152 | for (BuiltinClinicData builtin : enclosingType.getValue()) {
|
155 | 153 | for (ArgumentClinicData arg : builtin.arguments) {
|
156 | 154 | imports.addAll(arg.imports);
|
| 155 | + if (arg.castNodeFactory != null) { |
| 156 | + imports.add("com.oracle.graal.python.nodes.function.PythonBuiltinBaseNode"); |
| 157 | + imports.add(CLINIC_PACKAGE + ".ArgumentCastNode"); |
| 158 | + } |
157 | 159 | }
|
158 | 160 | }
|
159 | 161 | for (String pkg : imports) {
|
|
0 commit comments