RetroSearch Browse

Showing content from https://github.com/doombeaker/oneflow/commit/1a96e882f56f0719a6b9f1c9c7461aaa640a9191 below:


 @@ -15,9 +15,12 @@ limitations under the License.

15 15

*/

16 16 17 17


 #if CUDA_VERSION >= 11000

18

-
 #define CUDA_SUPPORT_CUFFT

18

+
 #define YCPRINTLOG printf("CUDA_VERSION >= 11000 is true, version: %d, runtime: %d\r\n", CUDA_VERSION, (CUDA_VERSION >= 11000));

19

+
 #else

20

+
 #define YCPRINTLOG printf("CUDA_VERSION >= 11000 is false, version: %d, runtime: %d\r\n", CUDA_VERSION, (CUDA_VERSION >= 11000));

19 21


 #endif

20 22 23

+
 #define CUDA_SUPPORT_CUFFT

21 24


 #ifdef CUDA_SUPPORT_CUFFT

22 25 23 26


 #include "oneflow/core/framework/framework.h"


 @@ -82,6 +85,7 @@ class StftGpuKernel final : public user_op::OpKernel {

82 85


 private:

83 86


 using user_op::OpKernel::Compute;

84 87


 void Compute(user_op::KernelComputeContext* ctx) const override {

88

+
 YCPRINTLOG;

85 89


 const user_op::Tensor* input = ctx->Tensor4ArgNameAndIndex("input", 0);

86 90


 user_op::Tensor* output = ctx->Tensor4ArgNameAndIndex("output", 0);

87 91


 user_op::Tensor* tmp_buffer = ctx->Tensor4ArgNameAndIndex("tmp_buffer", 0);

RetroSearch is an open source project built by @garambo | Open a GitHub Issue

Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo

HTML: 3.2 | Encoding: UTF-8 | Version: 0.7.4