Просмотр исходного кода

TSL: Use correct elementType with WorkgroupInfoNode (#30222)

Renaud Rohlinger 1 год назад
Родитель
Сommit
5bad3350b2
1 измененных файлов с 19 добавлено и 0 удалено
  1. 19 0
      src/nodes/gpgpu/WorkgroupInfoNode.js

+ 19 - 0
src/nodes/gpgpu/WorkgroupInfoNode.js

@@ -104,6 +104,13 @@ class WorkgroupInfoNode extends Node {
 		 */
 		this.isWorkgroupInfoNode = true;
 
+		/**
+		 * The data type of the array buffer.
+		 *
+		 * @type {String}
+		 */
+		this.elementType = bufferType;
+
 		/**
 		 * TODO.
 		 *
@@ -141,6 +148,18 @@ class WorkgroupInfoNode extends Node {
 
 	}
 
+
+	/**
+	 * The data type of the array buffer.
+	 *
+	 * @return {String} The element type.
+	 */
+	getElementType() {
+
+		return this.elementType;
+
+	}
+
 	/**
 	 * Overwrites the default implementation since the input type
 	 * is inferred from the scope.

粤ICP备19079148号