Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .travis.yml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# Use container-based infrastructure
os: linux
dist: focal
dist: xenial

# Set default python env
# because the xgboost-spark library when running training code, it will
Expand All @@ -24,7 +24,7 @@ language: scala
scala:
- 2.12.13
jdk:
- openjdk11
- openjdk8

jobs:
include:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,6 @@
import java.lang.reflect.Method;
import java.nio.ByteBuffer;

import java.lang.ref.Cleaner;
import sun.misc.Unsafe;

public final class Platform {
Expand Down Expand Up @@ -146,32 +145,7 @@ public static long reallocateMemory(long address, long oldSize, long newSize) {
freeMemory(address);
return newMemory;
}

/**
* Uses internal JDK APIs to allocate a DirectByteBuffer while ignoring the JVM's
* MaxDirectMemorySize limit (the default limit is too low and we do not want to require users
* to increase it).
*/
@SuppressWarnings("unchecked")
public static ByteBuffer allocateDirectBuffer(int size) {
try {
Class<?> cls = Class.forName("java.nio.DirectByteBuffer");
Constructor<?> constructor = cls.getDeclaredConstructor(Long.TYPE, Integer.TYPE);
constructor.setAccessible(true);
Field cleanerField = cls.getDeclaredField("cleaner");
cleanerField.setAccessible(true);
final long memory = allocateMemory(size);
ByteBuffer buffer = (ByteBuffer) constructor.newInstance(memory, size);
Cleaner cleaner = Cleaner.create();
cleaner.register(buffer, () -> freeMemory(memory));
cleanerField.set(buffer, cleaner);
return buffer;
} catch (Exception e) {
throwException(e);
}
throw new IllegalStateException("unreachable");
}


public static void setMemory(long address, byte value, long size) {
_UNSAFE.setMemory(address, size, value);
}
Expand Down