1 files changed, 10 insertions, 0 deletions
diff --git a/gradle/src/main/groovy/com/android/build/gradle/internal/api/BaseVariantImpl.java b/gradle/src/main/groovy/com/android/build/gradle/internal/api/BaseVariantImpl.java
index e49b7c0..94ee0ab 100644
--- a/gradle/src/main/groovy/com/android/build/gradle/internal/api/BaseVariantImpl.java
+++ b/gradle/src/main/groovy/com/android/build/gradle/internal/api/BaseVariantImpl.java
@@ -137,4 +137,14 @@ abstract class BaseVariantImpl implements BaseVariant {
     public Task getAssemble() {
         return getVariantData().assembleTask;
     }
+
+    @Override
+    public void addGeneratedSourceFolders(@NonNull Task task, @NonNull File... sourceFolders) {
+        getVariantData().addGeneratedSourceFolders(task, sourceFolders);
+    }
+
+    @Override
+    public void addGeneratedSourceFolders(@NonNull Task task, @NonNull Iterable<File> sourceFolders) {
+        getVariantData().addGeneratedSourceFolders(task, sourceFolders);
+    }
 }