Procházet zdrojové kódy

- add native Fortran routine to put codelet structures in variable buffer mode

Olivier Aumage před 8 roky
rodič
revize
2d8c014cf9
2 změnil soubory, kde provedl 10 přidání a 0 odebrání
  1. 5 0
      include/fstarpu_mod.f90
  2. 5 0
      src/util/fstarpu.c

+ 5 - 0
include/fstarpu_mod.f90

@@ -664,6 +664,11 @@ module fstarpu_mod
                         type(c_ptr), value, intent(in) :: mode ! C function expects an intptr_t
                 end subroutine fstarpu_codelet_add_buffer
 
+                subroutine fstarpu_codelet_set_variable_nbuffers (cl) bind(C)
+                        use iso_c_binding, only: c_ptr,c_int
+                        type(c_ptr), value, intent(in) :: cl
+                end subroutine fstarpu_codelet_set_variable_nbuffers
+
                 subroutine fstarpu_codelet_set_nbuffers (cl, nbuffers) bind(C)
                         use iso_c_binding, only: c_ptr,c_int
                         type(c_ptr), value, intent(in) :: cl

+ 5 - 0
src/util/fstarpu.c

@@ -304,6 +304,11 @@ void fstarpu_codelet_add_buffer(struct starpu_codelet *cl, intptr_t _mode)
 	}
 }
 
+void fstarpu_codelet_set_variable_nbuffers(struct starpu_codelet *cl)
+{
+	cl->nbuffers = STARPU_VARIABLE_NBUFFERS;
+}
+
 void fstarpu_codelet_set_nbuffers(struct starpu_codelet *cl, int nbuffers)
 {
 	if (nbuffers >= 0)