• Home
• ai.h2o
• h2o-ext-xgboost
• 3.22.1.3
• source code
• XGBoostGPULock.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

hex.tree.xgboost.XGBoostGPULock Maven / Gradle / Ivy

package hex.tree.xgboost;

import java.util.HashMap;
import java.util.Map;

class XGBoostGPULock {

    private static final Map LOCKS = new HashMap<>();

    static XGBoostGPULock lock(int gpuId) {
        if(!LOCKS.containsKey(gpuId)) {
            synchronized (XGBoostGPULock.class) {
                if(!LOCKS.containsKey(gpuId)) {
                    LOCKS.put(gpuId, new XGBoostGPULock());
                }
            }
        }
        return LOCKS.get(gpuId);
    }

}