-
Notifications
You must be signed in to change notification settings - Fork 5
/
darknet.Build.cs
68 lines (53 loc) · 2.16 KB
/
darknet.Build.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
// Fill out your copyright notice in the Description page of Project Settings.
using UnrealBuildTool;
using System.IO;
using System;
public class darknet : ModuleRules
{
protected string ThirdPartyPath
{
get { return Path.GetFullPath(Path.Combine(ModuleDirectory, "..", "..", "ThirdParty")); }
}
protected string CUDAPath
{
get { return System.Environment.GetEnvironmentVariable("CUDA_PATH"); }
}
public darknet(TargetInfo Target)
{
PrivateIncludePaths.AddRange(new string[]
{
Path.Combine(ThirdPartyPath),
});
PublicDependencyModuleNames.AddRange(new string[] { "Core", "CoreUObject", "Engine", "InputCore", "V8" });
PrivateDependencyModuleNames.AddRange(new string[] { });
LoadDarknet(Target);
}
private bool LoadDarknet(TargetInfo Target)
{
if ((Target.Platform == UnrealTargetPlatform.Win64) || (Target.Platform == UnrealTargetPlatform.Win32))
{
string LibrariesPath = Path.Combine(ThirdPartyPath, "darknet", "darknet");
if (Target.Platform == UnrealTargetPlatform.Win64)
{
LibrariesPath = Path.Combine(LibrariesPath, "x64");
}
if (Target.Configuration == UnrealTargetConfiguration.Debug)
{
LibrariesPath = Path.Combine(LibrariesPath, "Debug");
}
else
{
LibrariesPath = Path.Combine(LibrariesPath, "Release");
}
PublicAdditionalLibraries.Add(Path.Combine(LibrariesPath, "darknet.lib"));
Definitions.Add(string.Format("WITH_DARKNET=1"));
PublicAdditionalLibraries.Add(Path.Combine(CUDAPath, "lib", "x64", "cudart.lib"));
PublicAdditionalLibraries.Add(Path.Combine(CUDAPath, "lib", "x64", "cuda.lib"));
PublicAdditionalLibraries.Add(Path.Combine(CUDAPath, "lib", "x64", "cublas.lib"));
PublicAdditionalLibraries.Add(Path.Combine(CUDAPath, "lib", "x64", "curand.lib"));
return true;
}
Definitions.Add(string.Format("WITH_DARKNET=0"));
return false;
}
}