jakirkham

import os

from dask.callbacks import Callback
from dask.dot import dot_graph


class Track(Callback):
    def __init__(self, path='dasks', save_every=1):
        self.path = path
        self.save_every = save_every

jakirkham

import datetime
import getopt
import os
import sys
import time

import pynvml

def print_mem(dev_count):
    for index in range(dev_count):

jakirkham

jakirkham

jakirkham

jakirkham

import asyncio
import time
import numpy as np
import cupy
import numba

import dask.array as da
from dask_cuda import DGX, LocalCUDACluster
from dask.distributed import Client, wait

jakirkham

--- ucx_info_b	2019-08-08 11:01:32.820637475 -0700
+++ ucx_from_source_info_b	2019-08-08 11:02:15.976537113 -0700
@@ -8,6 +8,7 @@
 #define HAVE_CUDA                 1
 #define HAVE_CUDA_H               1
 #define HAVE_CUDA_RUNTIME_H       1
+#define HAVE_DC_EXP               1
 #define HAVE_DECL_ASPRINTF        1
 #define HAVE_DECL_BASENAME        1
 #define HAVE_DECL_CPU_ISSET       1

jakirkham

jakirkham

jakirkham