Skip to content

Commit 6e98ba7

Browse files
committed
[refactor] clean some warnings
1 parent c879342 commit 6e98ba7

File tree

3 files changed

+13
-11
lines changed

3 files changed

+13
-11
lines changed

tensorflow_recommenders_addons/dynamic_embedding/core/kernels/cuckoo_hashtable_op_gpu.cu.cc

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,6 @@ class CuckooHashTableOfTensorsGpu final : public LookupInterface {
115115
const Tensor& default_value) override {
116116
size_t len = d_keys.flat<K>().size();
117117
bool* d_status;
118-
gpu::ValueArrayBase<V>* d_default_value;
119118

120119
auto value_flat = value->flat_inner_dims<V, 2>();
121120
const auto default_flat = default_value.flat<V>();
@@ -140,7 +139,6 @@ class CuckooHashTableOfTensorsGpu final : public LookupInterface {
140139
CUDA_CHECK(cudaStreamSynchronize(_stream));
141140
}
142141
CUDA_CHECK(cudaFree(d_status));
143-
CUDA_CHECK(cudaFree(d_default_value));
144142
CUDA_CHECK(cudaStreamDestroy(_stream));
145143
}
146144
return Status::OK();
@@ -150,7 +148,6 @@ class CuckooHashTableOfTensorsGpu final : public LookupInterface {
150148
Tensor* value, const Tensor& default_value,
151149
Tensor* exists) {
152150
size_t len = d_keys.flat<K>().size();
153-
gpu::ValueArrayBase<V>* d_default_value;
154151

155152
auto value_flat = value->flat_inner_dims<V, 2>();
156153
const auto default_flat = default_value.flat<V>();
@@ -173,7 +170,6 @@ class CuckooHashTableOfTensorsGpu final : public LookupInterface {
173170
_stream, is_full_default);
174171
CUDA_CHECK(cudaStreamSynchronize(_stream));
175172
}
176-
CUDA_CHECK(cudaFree(d_default_value));
177173
CUDA_CHECK(cudaStreamDestroy(_stream));
178174
}
179175
return Status::OK();

tensorflow_recommenders_addons/dynamic_embedding/core/kernels/lookup_impl/lookup_table_op_cpu.h

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -115,20 +115,26 @@ class TableWrapperBase {
115115
public:
116116
virtual ~TableWrapperBase() {}
117117
virtual bool insert_or_assign(K key, ConstTensor2D<V>& value_flat,
118-
int64 value_dim, int64 index) {}
118+
int64 value_dim, int64 index) {
119+
return false;
120+
}
119121
virtual bool insert_or_accum(K key, ConstTensor2D<V>& value_or_delta_flat,
120-
bool exist, int64 value_dim, int64 index) {}
122+
bool exist, int64 value_dim, int64 index) {
123+
return false;
124+
}
121125
virtual void find(const K& key, Tensor2D<V>& value_flat,
122126
ConstTensor2D<V>& default_flat, int64 value_dim,
123127
bool is_full_size_default, int64 index) const {}
124128
virtual void find(const K& key, Tensor2D<V>& value_flat,
125129
ConstTensor2D<V>& default_flat, bool& exist,
126130
int64 value_dim, bool is_full_size_default,
127131
int64 index) const {}
128-
virtual size_t size() const {}
132+
virtual size_t size() const { return 0; }
129133
virtual void clear() {}
130-
virtual bool erase(const K& key) {}
131-
virtual Status export_values(OpKernelContext* ctx, int64 value_dim) {}
134+
virtual bool erase(const K& key) { return false; }
135+
virtual Status export_values(OpKernelContext* ctx, int64 value_dim) {
136+
return Status::OK();
137+
}
132138
};
133139

134140
template <class K, class V, size_t DIM>

tensorflow_recommenders_addons/dynamic_embedding/core/kernels/lookup_impl/lookup_table_op_gpu.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,8 +63,8 @@ class TableWrapperBase {
6363
virtual void get(const K* d_keys, ValueType<V>* d_vals, bool* d_status,
6464
size_t len, ValueType<V>* d_def_val, cudaStream_t stream,
6565
bool is_full_size_default) const {}
66-
virtual size_t get_size(cudaStream_t stream) const {}
67-
virtual size_t get_capacity() const {}
66+
virtual size_t get_size(cudaStream_t stream) const { return 0; }
67+
virtual size_t get_capacity() const { return 0; }
6868
virtual void remove(const K* d_keys, size_t len, cudaStream_t stream) {}
6969
virtual void clear(cudaStream_t stream) {}
7070
};

0 commit comments

Comments
 (0)