Skip to content

Commit

Permalink
Revert "testing new distr"
Browse files Browse the repository at this point in the history
This reverts commit ccbba42.
  • Loading branch information
ronpandolfi committed Sep 25, 2023
1 parent 71974f4 commit f28bd35
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 14 deletions.
3 changes: 1 addition & 2 deletions fvgp/gp.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,8 +19,7 @@
from loguru import logger
from .mcmc import mcmc
from hgdl.hgdl import HGDL
#from .gp2Scale import gp2Scale as gp2S
from .gp2ScaleR import gp2Scale as gp2S
from .gp2Scale import gp2Scale as gp2S
from dask.distributed import Variable
from dask.distributed import Client
from .sparse_matrix import gp2ScaleSparseMatrix
Expand Down
12 changes: 0 additions & 12 deletions fvgp/gp2ScaleR.py
Original file line number Diff line number Diff line change
Expand Up @@ -165,19 +165,7 @@ def _init_dask_client(self,dask_client):
print("Scheduler Address: ", dask_client.scheduler_info()["address"])
return dask_client, compute_worker_set,actor_worker

def submit_kernel_function(self, ij, hyperparameters, client):
i, j = ij
beg_i = i * self.batch_size
end_i = min((i+1) * self.batch_size, self.point_number)
beg_j = j * self.batch_size
end_j = min((j+1) * self.batch_size, self.point_number)

data = {"scattered_data": self.scatter_future, "hps": hyperparameters, "kernel": self.kernel,
"range_i": (beg_i, end_i), "range_j": (beg_j, end_j), "mode": "prior", "gpu": 0}

future = client.submit(kernel_function, data)

return future
#########################################################################
#########################################################################
#########################################################################
Expand Down

0 comments on commit f28bd35

Please sign in to comment.