tools: added cuda_library and cuda_binary rules

WORKSPACE.in

@@ -157,7 +157,7 @@ new_local_repository(
 new_local_repository(
     name = "cuda",
     build_file = "third_party/cuda.BUILD",
-    path = "/usr/local/cuda/include",
+    path = "/usr/local/cuda",
 )
 
 # Local-integ
@@ -194,12 +194,12 @@ http_archive(
 new_local_repository(
     name = "jsoncpp",
     build_file = "third_party/jsoncpp.BUILD",
-    path = "/usr/local/apollo/jsoncpp/"
+    path = "/usr/local/apollo/jsoncpp/",
 )
 
 #adv_plat
 new_local_repository(
     name = "adv_plat",
     build_file = "third_party/adv_plat.BUILD",
-    path = "/usr/local/apollo/adv_plat"
+    path = "/usr/local/apollo/adv_plat",
 )

third_party/cuda.BUILD

@@ -2,9 +2,11 @@ package(default_visibility = ["//visibility:public"])
 
 licenses(["notice"])
 
+
+
 cc_library(
     name = "cuda",
-    includes = ["."],
+    includes = ["include"],
     linkopts = [
         "-L/usr/local/cuda/lib64",
         "-lOpenCL",

tools/cuda_library.bzl

+cuda_srcs = FileType([
+    ".cu",
+    ".cc",
+    ".cpp",
+])
+
+cuda_headers = FileType([
+    ".h",
+    ".hpp",
+])
+
+cuda_arch = [
+    "-arch=sm_30",
+    "-gencode=arch=compute_30,code=sm_30",
+    "-gencode=arch=compute_50,code=sm_50",
+    "-gencode=arch=compute_52,code=sm_52",
+    "-gencode=arch=compute_60,code=sm_60",
+    "-gencode=arch=compute_61,code=sm_61",
+    "-gencode=arch=compute_61,code=compute_61",
+]
+
+def cuda_library_impl(ctx):
+    s = set()
+    flags = ' '.join(ctx.attr.flags)
+    output = ctx.outputs.out
+    lib_flags = ["--shared", "--compiler-options '-fPIC'"]
+    args = ctx.attr.flags + [f.path for f in ctx.files.srcs] + [f.path for f in ctx.files.hdrs] + [f.path for f in ctx.files.deps]
+    ctx.actions.run_shell(
+            inputs=ctx.files.srcs + ctx.files.hdrs,
+            outputs=[ctx.outputs.out],
+            arguments=args,
+            env={ 'PATH':'/usr/local/cuda/bin:/usr/local/bin:/usr/bin:/bin', },
+            command="/usr/local/cuda/bin/nvcc %s %s -o %s %s" % (' '.join(cuda_arch), ' '.join(lib_flags), output.path, " ".join(args))
+     )
+
+def cuda_binary_impl(ctx):
+    flags = ' '.join(ctx.attr.flags)
+    args = ctx.attr.flags + [f.path for f in ctx.files.srcs] + [f.path for f in ctx.files.hdrs] + [f.path for f in ctx.attr.deps]
+    output = ctx.outputs.out
+    ctx.actions.run_shell(
+            inputs=ctx.files.srcs + ctx.files.hdrs,
+            outputs=[ctx.outputs.out],
+            arguments=args,
+            env={ 'PATH':'/usr/local/cuda/bin:/usr/local/bin:/usr/bin:/bin', },
+            command="/usr/local/cuda/bin/nvcc %s -o %s %s" % (' '.join(cuda_arch), output.path, " ".join(args)),
+     )
+
+cuda_library = rule(
+    attrs = {
+        "hdrs": attr.label_list(allow_files = cuda_headers),
+        "srcs": attr.label_list(
+            allow_files = cuda_srcs,
+            mandatory = True,
+        ),
+        "deps": attr.label_list(allow_files = False),
+        "flags": attr.label_list(allow_files = False),
+    },
+    outputs = {"out": "lib%{name}.so"},
+    implementation = cuda_library_impl,
+)
+
+cuda_binary = rule(
+    attrs = {
+        "hdrs": attr.label_list(allow_files = cuda_headers),
+        "srcs": attr.label_list(
+            allow_files = cuda_srcs,
+            mandatory = True,
+        ),
+        "deps": attr.label_list(allow_files = False),
+        "flags": attr.label_list(allow_files = False),
+    },
+    executable = True,
+    outputs = {"out": "%{name}"},
+    implementation = cuda_binary_impl,
+)