Commit 25d7fde8 authored by gaoqiong's avatar gaoqiong
Browse files

lite

parent 8439d29f
......@@ -192,7 +192,7 @@ void BinaryElementWiseNoBroadcastImpl(
#ifdef USE_ROCM
const int num_elements_per_thread = 2;
const int num_threads_per_block = 512;
const int num_threads_per_block = 256;
#else
const int num_elements_per_thread = GridDim::maxElementsPerThread;
const int num_threads_per_block = GridDim::maxThreadsPerBlock;
......@@ -228,7 +228,7 @@ void BinaryElementWiseImpl(
#ifdef USE_ROCM
const int num_elements_per_thread = 2;
const int num_threads_per_block = 512;
const int num_threads_per_block = 256;
#else
const int num_elements_per_thread = GridDim::maxElementsPerThread;
const int num_threads_per_block = GridDim::maxThreadsPerBlock;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment