jitcode.cc 1.1 KB
Newer Older
T
tensor-tang 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License. */

T
tensor-tang 已提交
15
#include "paddle/fluid/operators/jit/gen/jitcode.h"
T
tensor-tang 已提交
16 17 18

namespace paddle {
namespace operators {
T
tensor-tang 已提交
19
namespace jit {
T
tensor-tang 已提交
20 21 22 23 24 25 26

template <>
size_t GetKey<int>(int d) {
  return d;
}

// template <>
T
tensor-tang 已提交
27
// std::shared_ptr<const GenBase> CreateJitCode<KernelType::vmul, int>(int attr)
T
tensor-tang 已提交
28 29
// {
//   if (UseJitCode<KernelType::vmul, int>(attr)) {
T
tensor-tang 已提交
30
//     return std::make_shared<gen::VMulJitCode<int>>(attr,
T
tensor-tang 已提交
31 32 33 34 35
//     CodeSize<KernelType::vmul, int>(attr)));
//   }
//   return nullptr;
// }

T
tensor-tang 已提交
36
}  // namespace jit
T
tensor-tang 已提交
37 38
}  // namespace operators
}  // namespace paddle