diff --git a/tools/amd_build/disabled_features.yaml b/tools/amd_build/disabled_features.yaml
index 9dd00f675b4acd..021f4187dcdf6d 100644
--- a/tools/amd_build/disabled_features.yaml
+++ b/tools/amd_build/disabled_features.yaml
@@ -81,6 +81,7 @@
           "struct curandStateMtgp32*": "curandStateMtgp32*",
           "__host__ void THCRandom_getRNGState": "extern \"C\" __host__ void THCRandom_getRNGState",
           "__host__ void THCRandom_setRNGState": "extern \"C\" __host__ void THCRandom_setRNGState",
+          "state[threadIdx.x].k = kernel;" : "state[threadIdx.x].set_params(kernel);"
         }
       },
       {
@@ -211,13 +212,6 @@
       "functions": [
         "THCTensor_(getTextureObject)"
       ]
-    },
-    {
-      "path": "aten/src/THC/THCTensorRandom.cu",
-      "functions": [
-        "THCRandom_setRNGState",
-        "set_rngstate_kernel"
-      ]
     }
   ]
 }