Load and init Extrae on all processes in example script (#18)

bsc-quantic · May 6, 2024 · 7e00057 · 7e00057
1 parent 188400d
commit 7e00057
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/scripts/test-distributed-work.jl b/scripts/test-distributed-work.jl
@@ -1,11 +1,11 @@
 using Distributed
-using Extrae
-
-ENV["JULIA_DEBUG"] = Extrae
-@everywhere ENV["JULIA_DEBUG"] = Extrae
 
 addprocs(1)
 
+@everywhere using Extrae
+
+@everywhere ENV["JULIA_DEBUG"] = Extrae
+
 function random_sleep()
     println("Worker started: ", myid())
     sleep(rand((1, 2, 3, 4, 5)))
@@ -18,7 +18,7 @@ end
 end
 
 function test_distributed_work()
-
+    @everywhere Extrae.init()
     A = rand(1000, 1000)
     a1 = @spawnat :any matrix_multiply(A)
     a2 = @spawnat :any matrix_multiply(A)
@@ -29,4 +29,4 @@ end
 
 test_distributed_work()
 
-println("END TEST")
+println("END TEST")