cuda - cudaMemcpyToSymbol 性能

Question

我有一些函数可以在常量设备内存中加载一个变量并启动一个内核函数。我注意到一个函数第一次在常量内存中加载一个变量需要 0.6 秒，但下一次在常量内存上的加载非常快（0.0008 秒）。无论哪个函数是主函数中的第一个，都会发生此行为。下面是一个示例代码：

        __constant__ double res1;

        __global__kernel1(...) {...}

        void function1() {
            double resHost = 255 / ((double) size);
            CUDA_CHECK_RETURN(cudaMemcpyToSymbol(res1, &resHost, sizeof(double)));


            //prepare and launch kernel
        }

        __constant__ double res2;

        __global__kernel2(...) {...}

        void function2() {
            double resHost = 255 / ((double) size);
            CUDA_CHECK_RETURN(cudaMemcpyToSymbol(res2, &resHost, sizeof(double)));


            //prepare and launch kernel
        }

        int main(){
            function1(); //takes 0.6 seconds for loading
            function2(); // takes 0.0008 seconds for loading
            function1(); //takes 0.0008 seconds for loading

            return 0;
        }

为什么会这样？我可以避免吗？

Answer 1

为什么会这样？

延迟运行时 API 上下文建立和设置。

我可以避免吗？

不会。第一次需要上下文的运行时 API 调用将导致显着的设置延迟，在您的情况下，这是第一次cudaMemcpyToSymbol调用。