• Olivier Delalleau's avatar
    Fix when forcing cuda to be disabled · 2b9ba848
    Olivier Delalleau 提交于
    The GpuOp class must be available even when Cuda is disabled (e.g. by
    forcing cuda_available = False). Otherwise some tests fail.
    
    This commit moves the definition of GpuOp directly into cuda/__init__.py
    so that it works.
    2b9ba848
multinomial.py 10.8 KB