modify version code

csrc/adam/custom_hip_kernel.hip

+// !!! This is a file automatically generated by hipify!!!
+#include "hip/hip_runtime.h"
+
+
+#include "custom_hip_layers.h"
+
+__global__ void param_update_kernel(const float* input, __half* output, int size)
+{
+    int id = blockIdx.x * blockDim.x + threadIdx.x;
+
+    if (id < size) { output[id] = (__half)input[id]; }
+}
+
+void launch_param_update(const float* input, __half* output, int size, hipStream_t stream)
+{
+    int threads = 1024;
+
+    dim3 grid_dim((size - 1) / threads + 1);
+    dim3 block_dim(threads);
+
+   hipLaunchKernelGGL(( param_update_kernel), dim3(grid_dim), dim3(block_dim), 0, stream, input, output, size);
+}

op_builder/builder.py

@@ -190,7 +190,10 @@ class OpBuilder(ABC):
         if OpBuilder.is_rocm_pytorch():
             from torch.utils.cpp_extension import ROCM_HOME
             #with open('/opt/rocm/.info/version-dev', 'r') as file:
-            with open('/opt/dtk-22.04/.info/version-dev', 'r') as file:
+        #aiss 20220810
+            ds_build_version = os.getenv('ROCM_PATH', "")
+           
+            with open('{0}/.info/version-dev'.format(ds_build_version), 'r') as file:
                 ROCM_VERSION_DEV_RAW = file.read()
             ROCM_MAJOR = ROCM_VERSION_DEV_RAW.split('.')[0]
             ROCM_MINOR = ROCM_VERSION_DEV_RAW.split('.')[1]

setup.py

@@ -178,7 +178,12 @@ if command_exists('git') and 'DS_BUILD_STRING' not in os.environ:
         git_hash = result.decode('utf-8').strip()
         result = subprocess.check_output(git_branch_cmd, shell=True)
         git_branch = result.decode('utf-8').strip()
-        git_hash +="dtk22.04"
+        ##aiss 20220810
+        if os.getenv('DS_BUILD_VERSION'):
+            version_dtk = os.getenv('DS_BUILD_VERSION', "")
+            git_hash += "." + version_dtk
+
+        #git_hash +="dtk22.04"
     except subprocess.CalledProcessError:
         git_hash = "unknown"
         git_branch = "unknown"